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

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.