public final class RecvTPUEmbeddingActivations extends PrimitiveOp implements Iterable<Operand<Float>>
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.
operation
Modifier and Type | Method and Description |
---|---|
static RecvTPUEmbeddingActivations |
create(Scope scope,
Long numOutputs,
String config)
Factory method to create a class wrapping a new RecvTPUEmbeddingActivations operation.
|
Iterator<Operand<Float>> |
iterator() |
List<Output<Float>> |
outputs()
A TensorList of embedding activations containing one Tensor per
embedding table in the model.
|
equals, hashCode, op, toString
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
forEach, spliterator
public static RecvTPUEmbeddingActivations create(Scope scope, Long numOutputs, String config)
scope
- current scopenumOutputs
- The number of output activation tensors, equal to the number of
embedding tables in the model.config
- Serialized TPUEmbeddingConfiguration proto.public List<Output<Float>> outputs()
Copyright © 2022. All rights reserved.