ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

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

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.