RecvTPUEmbeddingActivations

public final class RecvTPUEmbeddingActivations

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
create ( Scope scope, Long numOutputs, String config)
Factory method to create a class wrapping a new RecvTPUEmbeddingActivations operation.
Iterator< Operand <Float>>
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 Iterator< Operand <Float>> iterator ()

public List< Output <Float>> outputs ()

A TensorList of embedding activations containing one Tensor per embedding table in the model.