public static class OutfeedDequeueTuple.Options extends Object
OutfeedDequeueTuple
Modifier and Type | Method and Description |
---|---|
OutfeedDequeueTuple.Options |
deviceOrdinal(Long deviceOrdinal) |
public OutfeedDequeueTuple.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.