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.
Constants
String | OP_NAME | The name of this op, as known by TensorFlow core engine |
Public Methods
static RecvTPUEmbeddingActivations | |
Iterator<Operand<TFloat32>> |
iterator()
|
List<Output<TFloat32>> |
outputs()
A TensorList of embedding activations containing one Tensor per
embedding table in the model.
|
Inherited Methods
Constants
public static final String OP_NAME
The name of this op, as known by TensorFlow core engine
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