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
Inherited Methods
boolean |
equals(Object arg0)
|
final Class<?> |
getClass()
|
int |
hashCode()
|
final void |
notify()
|
final void |
notifyAll()
|
String |
toString()
|
final void |
wait(long arg0, int arg1)
|
final void |
wait(long arg0)
|
final void |
wait()
|
abstract ExecutionEnvironment |
env()
Return the execution environment this op was created in.
|
abstract Operation |
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