public static class
DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.Options
Optional attributes for DynamicEnqueueTPUEmbeddingArbitraryTensorBatch
Public Methods
| DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.Options |
combiners(List<String> combiners)
|
Inherited Methods
| boolean |
equals(Object arg0)
|
| final Class<?> |
getClass()
|
| int |
hashCode()
|
| final void |
notify()
|
| final void |
notifyAll()
|
| String |
toString()
|
| final void |
wait(long arg0, int arg1)
|
| final void |
wait(long arg0)
|
| final void |
wait()
|
Public Methods
public DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.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. |
|---|