public static class EnqueueTPUEmbeddingSparseTensorBatch.Options extends Object
EnqueueTPUEmbeddingSparseTensorBatch
Modifier and Type | Method and Description |
---|---|
EnqueueTPUEmbeddingSparseTensorBatch.Options |
combiners(List<String> combiners) |
EnqueueTPUEmbeddingSparseTensorBatch.Options |
deviceOrdinal(Long deviceOrdinal) |
EnqueueTPUEmbeddingSparseTensorBatch.Options |
maxSequenceLengths(List<Long> maxSequenceLengths) |
public EnqueueTPUEmbeddingSparseTensorBatch.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 EnqueueTPUEmbeddingSparseTensorBatch.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 EnqueueTPUEmbeddingSparseTensorBatch.Options maxSequenceLengths(List<Long> maxSequenceLengths)
maxSequenceLengths
- Copyright © 2022. All rights reserved.