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

InfeedEnqueue.Options

public static class InfeedEnqueue.Options

Optional attributes for InfeedEnqueue

Public Methods

InfeedEnqueue.Options
deviceOrdinal(Long deviceOrdinal)
InfeedEnqueue.Options
layout(List<Long> layout)
InfeedEnqueue.Options
shape(Shape shape)

Inherited Methods

Public Methods

public 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 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 InfeedEnqueue.Options shape (Shape shape)

Parameters
shape The shape of the tensor.