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