public static class InfeedEnqueueTuple.Options extends Object
InfeedEnqueueTuple
Modifier and Type | Method and Description |
---|---|
InfeedEnqueueTuple.Options |
deviceOrdinal(Long deviceOrdinal) |
InfeedEnqueueTuple.Options |
layouts(List<Long> layouts) |
public InfeedEnqueueTuple.Options layouts(List<Long> layouts)
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.public InfeedEnqueueTuple.Options deviceOrdinal(Long deviceOrdinal)
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.Copyright © 2022. All rights reserved.