InfeedEnqueue

public final class InfeedEnqueue

An op which feeds a single Tensor value into the computation.

Nested Classes

class InfeedEnqueue.Options Optional attributes for InfeedEnqueue  

Public Methods

static <T> InfeedEnqueue
create(Scope scope, Operand<T> input, Options... options)
Factory method to create a class wrapping a new InfeedEnqueue operation.
static InfeedEnqueue.Options
deviceOrdinal(Long deviceOrdinal)
static InfeedEnqueue.Options
layout(List<Long> layout)
static InfeedEnqueue.Options
shape(Shape shape)

Inherited Methods

org.tensorflow.op.PrimitiveOp
final boolean
equals(Object obj)
final int
hashCode()
Operation
op()
Returns the underlying Operation
final String
toString()
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 InfeedEnqueue create (Scope scope, Operand<T> input, Options... options)

Factory method to create a class wrapping a new InfeedEnqueue operation.

Parameters
scope current scope
input A tensor that will be provided using the infeed mechanism.
options carries optional attributes values
Returns
  • a new instance of InfeedEnqueue

public static InfeedEnqueue.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.

public static InfeedEnqueue.Options layout (List<Long> layout)

Parameters
layout A vector holding the requested layout in minor-to-major sequence. If a layout attribute is passed, but its values are all -1, the layout will be computed by the infeed operation.

public static InfeedEnqueue.Options shape (Shape shape)

Parameters
shape The shape of the tensor.