org.tensorflow.op.tpu

Class EnqueueTPUEmbeddingSparseBatch.Options

java.lang.Object

org.tensorflow.op.tpu.EnqueueTPUEmbeddingSparseBatch.Options

Enclosing class:

EnqueueTPUEmbeddingSparseBatch

public static class EnqueueTPUEmbeddingSparseBatch.Options
extends Object

Optional attributes for EnqueueTPUEmbeddingSparseBatch

Method Summary

Modifier and Type	Method and Description
EnqueueTPUEmbeddingSparseBatch.Options	combiners(List<String> combiners) Sets the combiners option.
EnqueueTPUEmbeddingSparseBatch.Options	combiners(String... combiners) Sets the combiners option.
EnqueueTPUEmbeddingSparseBatch.Options	deviceOrdinal(Long deviceOrdinal) Sets the deviceOrdinal option.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

deviceOrdinal

public EnqueueTPUEmbeddingSparseBatch.Options deviceOrdinal(Long deviceOrdinal)

Sets the deviceOrdinal option.

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.

Returns:

this Options instance.

combiners

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

Sets the combiners option.

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.

Returns:

this Options instance.

combiners

public EnqueueTPUEmbeddingSparseBatch.Options combiners(String... combiners)

Sets the combiners option.

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.

Returns:

this Options instance.