public static class InfeedEnqueue.Options extends Object
InfeedEnqueue
Modifier and Type | Method and Description |
---|---|
InfeedEnqueue.Options |
deviceOrdinal(Long deviceOrdinal) |
InfeedEnqueue.Options |
layout(List<Long> layout) |
InfeedEnqueue.Options |
shape(Shape shape) |
public InfeedEnqueue.Options shape(Shape shape)
shape
- The shape of the tensor.public InfeedEnqueue.Options layout(List<Long> layout)
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 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.