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