org.tensorflow.op.core

Class InfeedEnqueueTuple.Options

java.lang.Object

org.tensorflow.op.core.InfeedEnqueueTuple.Options

Enclosing class:

InfeedEnqueueTuple

public static class InfeedEnqueueTuple.Options
extends Object

Optional attributes for InfeedEnqueueTuple

Method Summary

Modifier and Type	Method and Description
InfeedEnqueueTuple.Options	deviceOrdinal(Long deviceOrdinal)
InfeedEnqueueTuple.Options	layouts(List<Long> layouts)

Methods inherited from class java.lang.Object

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

Method Detail

layouts

public InfeedEnqueueTuple.Options layouts(List<Long> layouts)

Parameters:

layouts - A vector holding the requested layout in minor-to-major sequence for all the tuple shapes, in the order the shapes appear in the "shapes" input. The layout elements for a sub-shape can be set to -1, in which case the corresponding layout will be computed by the infeed operation.

deviceOrdinal

public InfeedEnqueueTuple.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.