An op that receives embedding activations on the TPU.
The TPU system performs the embedding lookups and aggregations specified by the arguments to TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor)Batch. The results of these aggregations are visible to the Tensorflow Graph as the outputs of a RecvTPUEmbeddingActivations op. This op returns a list containing one Tensor of activations per table specified in the model. There can be at most one RecvTPUEmbeddingActivations op in the TPU graph.
Public Methods
static RecvTPUEmbeddingActivations | |
Iterator< Operand <Float>> |
iterator
()
|
List< Output <Float>> |
outputs
()
A TensorList of embedding activations containing one Tensor per
embedding table in the model.
|
Inherited Methods
Public Methods
public static RecvTPUEmbeddingActivations create ( Scope scope, Long numOutputs, String config)
Factory method to create a class wrapping a new RecvTPUEmbeddingActivations operation.
Parameters
scope | current scope |
---|---|
numOutputs | The number of output activation tensors, equal to the number of embedding tables in the model. |
config | Serialized TPUEmbeddingConfiguration proto. |
Returns
- a new instance of RecvTPUEmbeddingActivations
public List< Output <Float>> outputs ()
A TensorList of embedding activations containing one Tensor per embedding table in the model.