An op computes the size of the deduplication data from embedding core and returns the updated config.
This op is to compute size of the deduplication data so to provide this information to the op that computes the tuple mask of deduplication data can have static output shape.
Public Methods
Output<Integer> |
asOutput()
Returns the symbolic handle of a tensor.
|
static ComputeDedupDataSize | |
Output<Integer> |
numElements()
The size of the deduplicated data from infeed.
|
Inherited Methods
Public Methods
public Output<Integer> asOutput ()
Returns the symbolic handle of a tensor.
Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.
public static ComputeDedupDataSize create (Scope scope, String config)
Factory method to create a class wrapping a new ComputeDedupDataSize operation.
Parameters
scope | current scope |
---|---|
config | Serialized TPUEmbeddingConfiguration proto. |
Returns
- a new instance of ComputeDedupDataSize