{ }
An op that enqueues TPUEmbedding input indices from a SparseTensor.
tf.raw_ops.EnqueueTPUEmbeddingSparseBatch(
sample_indices,
embedding_indices,
aggregation_weights,
mode_override,
device_ordinal=-1,
combiners=[],
name=None
)
This Op eases the porting of code that uses embedding_lookup_sparse(), although some Python preprocessing of the SparseTensor arguments to embedding_lookup_sparse() is required to produce the arguments to this Op, since only a single EnqueueTPUEmbeddingSparseBatch Op is allowed per training step.
The tensors at corresponding positions in the three input lists must have the same shape, i.e. rank 1 with dim_size() equal to the total number of lookups into the table described by the corresponding table_id.
Returns | |
---|---|
The created Operation. |