ComputeDedupDataSize

public final class ComputeDedupDataSize

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
create(Scope scope, String config)
Factory method to create a class wrapping a new ComputeDedupDataSize operation.
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

public Output<Integer> numElements ()

The size of the deduplicated data from infeed.