Warning: This API is deprecated and will be removed in a future version of TensorFlow after the replacement is stable.

InfeedEnqueue.Options

Stay organized with collections Save and categorize content based on your preferences.
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.