public static class EnqueueTPUEmbeddingIntegerBatch.Options extends Object
EnqueueTPUEmbeddingIntegerBatch
Modifier and Type | Method and Description |
---|---|
EnqueueTPUEmbeddingIntegerBatch.Options |
deviceOrdinal(Long deviceOrdinal) |
public EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal(Long deviceOrdinal)
deviceOrdinal
- The TPU device to use. Should be >= 0 and less than the number
of TPU cores in the task on which the node is placed.Copyright © 2022. All rights reserved.