EnqueueTPUEmbeddingRaggedTensorBatch.Options

public static class EnqueueTPUEmbeddingRaggedTensorBatch.Options

Optional attributes for EnqueueTPUEmbeddingRaggedTensorBatch

Public Methods

EnqueueTPUEmbeddingRaggedTensorBatch.Options
combiners(List<String> combiners)
EnqueueTPUEmbeddingRaggedTensorBatch.Options
deviceOrdinal(Long deviceOrdinal)
EnqueueTPUEmbeddingRaggedTensorBatch.Options
maxSequenceLengths(List<Long> maxSequenceLengths)

Inherited Methods

Public Methods

public EnqueueTPUEmbeddingRaggedTensorBatch.Options combiners (List<String> combiners)

Parameters
combiners A list of string scalars, one for each embedding table that specify how to normalize the embedding activations after weighted summation. Supported combiners are 'mean', 'sum', or 'sqrtn'. It is invalid to have the sum of the weights be 0 for 'mean' or the sum of the squared weights be 0 for 'sqrtn'. If combiners isn't passed, the default is to use 'sum' for all tables.

public EnqueueTPUEmbeddingRaggedTensorBatch.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.

public EnqueueTPUEmbeddingRaggedTensorBatch.Options maxSequenceLengths (List<Long> maxSequenceLengths)