An op enabling differentiation of TPU Embeddings.
tf.raw_ops.TPUEmbeddingActivations(
embedding_variable, sliced_activations, table_id, lookup_id, name=None
)
This op simply returns its first input, which is assumed to have been sliced from the Tensors returned by TPUEmbeddingDequeueActivations. The presence of this op, and its first argument being a trainable Variable, enables automatic differentiation of graphs containing embeddings via the TPU Embedding Python libraries.
Returns | |
---|---|
A Tensor of type float32 .
|