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.

Constants

String OP_NAME The name of this op, as known by TensorFlow core engine

Public Methods

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

Constant Value: "RecvTPUEmbeddingActivations"

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<TFloat32>> iterator ()

public List<Output<TFloat32>> outputs ()

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