public static class
EnqueueTPUEmbeddingIntegerBatch.Options
Optional attributes for EnqueueTPUEmbeddingIntegerBatch
Public Methods
EnqueueTPUEmbeddingIntegerBatch.Options |
deviceOrdinal(Long deviceOrdinal)
|
Inherited Methods
Public Methods
public EnqueueTPUEmbeddingIntegerBatch.Options deviceOrdinal (Long deviceOrdinal)
Parameters
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. |
---|