InfeedEnqueuePrelinearizedBuffer

public final class InfeedEnqueuePrelinearizedBuffer

An op which enqueues prelinearized buffer into TPU infeed.

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

org.tensorflow.op.PrimitiveOp
final boolean
equals(Object obj)
final int
Operation
op()
Returns the underlying Operation
final String
boolean
equals(Object arg0)
final Class<?>
getClass()
int
hashCode()
final void
notify()
final void
notifyAll()
String
toString()
final void
wait(long arg0, int arg1)
final void
wait(long arg0)
final void
wait()

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.