EnqueueTPUEmbeddingSparseTensorBatch.Options

public static class EnqueueTPUEmbeddingSparseTensorBatch.Options

Optional attributes for EnqueueTPUEmbeddingSparseTensorBatch

Public Methods

EnqueueTPUEmbeddingSparseTensorBatch.Options	combiners(List<String> combiners)
EnqueueTPUEmbeddingSparseTensorBatch.Options	deviceOrdinal(Long deviceOrdinal)
EnqueueTPUEmbeddingSparseTensorBatch.Options	maxSequenceLengths(List<Long> maxSequenceLengths)
EnqueueTPUEmbeddingSparseTensorBatch.Options	numFeatures(List<Long> numFeatures)

Inherited Methods

From class java.lang.Object

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 EnqueueTPUEmbeddingSparseTensorBatch.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 EnqueueTPUEmbeddingSparseTensorBatch.Options deviceOrdinal (Long deviceOrdinal)

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.

public EnqueueTPUEmbeddingSparseTensorBatch.Options maxSequenceLengths (List<Long> maxSequenceLengths)

public EnqueueTPUEmbeddingSparseTensorBatch.Options numFeatures (List<Long> numFeatures)