InfeedEnqueue.Options

public static class InfeedEnqueue.Options

Optional attributes for InfeedEnqueue

Public Methods

Inherited Methods

boolean
equals(Object arg0)
final Class<?>
getClass()
int
hashCode()
final void
notify()
final void
notifyAll()
String
toString()
final void
wait(long arg0, int arg1)
final void
wait(long arg0)
final void
wait()

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.