public static class EnqueueTPUEmbeddingRaggedTensorBatch.Options extends Object
EnqueueTPUEmbeddingRaggedTensorBatch
Modifier and Type | Method and Description |
---|---|
EnqueueTPUEmbeddingRaggedTensorBatch.Options |
combiners(List<String> combiners)
Sets the combiners option.
|
EnqueueTPUEmbeddingRaggedTensorBatch.Options |
combiners(String... combiners)
Sets the combiners option.
|
EnqueueTPUEmbeddingRaggedTensorBatch.Options |
deviceOrdinal(Long deviceOrdinal)
Sets the deviceOrdinal option.
|
EnqueueTPUEmbeddingRaggedTensorBatch.Options |
maxSequenceLengths(List<Long> maxSequenceLengths)
Sets the maxSequenceLengths option.
|
EnqueueTPUEmbeddingRaggedTensorBatch.Options |
maxSequenceLengths(Long... maxSequenceLengths)
Sets the maxSequenceLengths option.
|
EnqueueTPUEmbeddingRaggedTensorBatch.Options |
numFeatures(List<Long> numFeatures)
Sets the numFeatures option.
|
EnqueueTPUEmbeddingRaggedTensorBatch.Options |
numFeatures(Long... numFeatures)
Sets the numFeatures option.
|
public EnqueueTPUEmbeddingRaggedTensorBatch.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.public EnqueueTPUEmbeddingRaggedTensorBatch.Options combiners(List<String> combiners)
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 combiners(String... combiners)
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 maxSequenceLengths(List<Long> maxSequenceLengths)
maxSequenceLengths
- the maxSequenceLengths optionpublic EnqueueTPUEmbeddingRaggedTensorBatch.Options maxSequenceLengths(Long... maxSequenceLengths)
maxSequenceLengths
- the maxSequenceLengths optionpublic EnqueueTPUEmbeddingRaggedTensorBatch.Options numFeatures(List<Long> numFeatures)
numFeatures
- the numFeatures optionpublic EnqueueTPUEmbeddingRaggedTensorBatch.Options numFeatures(Long... numFeatures)
numFeatures
- the numFeatures optionCopyright © 2015–2022. All rights reserved.