ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

EnqueueTPUEmbeddingIntegerBatch.Options

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.