public static class
EnqueueTPUEmbeddingBatch.Options
Optional attributes for EnqueueTPUEmbeddingBatch
Public Methods
| EnqueueTPUEmbeddingBatch.Options |
combiners(List<String> combiners)
|
| EnqueueTPUEmbeddingBatch.Options |
deviceOrdinal(Long deviceOrdinal)
|
Inherited Methods
Public Methods
public EnqueueTPUEmbeddingBatch.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 EnqueueTPUEmbeddingBatch.Options deviceOrdinal (Long deviceOrdinal)
Parameters
| deviceOrdinal | The TPU device to use. This should be -1 when the Op is running on a TPU device, and >= 0 when the Op is running on the CPU device. |
|---|