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.