InfeedEnqueuePrelinearizedBuffer

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.