public final class
InfeedEnqueuePrelinearizedBuffer
An op which enqueues prelinearized buffer into TPU infeed.
Nested Classes
| class | InfeedEnqueuePrelinearizedBuffer.Options |
Optional attributes for
InfeedEnqueuePrelinearizedBuffer
|
|
Public Methods
| static InfeedEnqueuePrelinearizedBuffer |
create
(
Scope
scope,
Operand
<?> input,
Options...
options)
Factory method to create a class wrapping a new InfeedEnqueuePrelinearizedBuffer operation.
|
| static InfeedEnqueuePrelinearizedBuffer.Options |
deviceOrdinal
(Long deviceOrdinal)
|
Inherited Methods
Public Methods
public static InfeedEnqueuePrelinearizedBuffer create ( Scope scope, Operand <?> input, Options... options)
Factory method to create a class wrapping a new InfeedEnqueuePrelinearizedBuffer operation.
Parameters
| scope | current scope |
|---|---|
| input | A variant tensor representing linearized output. |
| options | carries optional attributes values |
Returns
- a new instance of InfeedEnqueuePrelinearizedBuffer
public static InfeedEnqueuePrelinearizedBuffer.Options deviceOrdinal (Long deviceOrdinal)
Parameters
| deviceOrdinal | The TPU device to use. This should be -1 when the Op is running on a TPU device and = 0 when the Op is running on the CPU device. |
|---|