org.tensorflow.op.core

Class InfeedEnqueue.Options

java.lang.Object

org.tensorflow.op.core.InfeedEnqueue.Options

Enclosing class:

InfeedEnqueue

public static class InfeedEnqueue.Options
extends Object

Optional attributes for InfeedEnqueue

Method Summary

Modifier and Type	Method and Description
InfeedEnqueue.Options	deviceOrdinal(Long deviceOrdinal)
InfeedEnqueue.Options	layout(List<Long> layout)
InfeedEnqueue.Options	shape(Shape shape)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

shape

public InfeedEnqueue.Options shape(Shape shape)

Parameters:

shape - The shape of the tensor.

layout

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.

deviceOrdinal

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.