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

From class java.lang.Object

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.