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. |
|---|