Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand

ComputeDedupDataTupleMask

public final class ComputeDedupDataTupleMask

An op computes tuple mask of deduplication data from embedding core.

The deduplication data receiving from embedding core is a Tensor with type=DT_VARIANT. The tensor itself is an XLA nested tuple, whose elements are rank 1 tensors. This op is to represents types and length of these elements.

Public Methods

Output<Integer>
asOutput()
Returns the symbolic handle of a tensor.
static ComputeDedupDataTupleMask
create(Scope scope, String config)
Factory method to create a class wrapping a new ComputeDedupDataTupleMask operation.
Output<Integer>
outputShape()
A 2-D int tensor represent mask of deduplication data tuple generated by `XlaRecvTPUEmbeddingDeduplicationData`.

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 ComputeDedupDataTupleMask create (Scope scope, String config)

Factory method to create a class wrapping a new ComputeDedupDataTupleMask operation.

Parameters
scope current scope
config Serialized TPUEmbeddingConfiguration proto.
Returns
  • a new instance of ComputeDedupDataTupleMask

public Output<Integer> outputShape ()

A 2-D int tensor represent mask of deduplication data tuple generated by `XlaRecvTPUEmbeddingDeduplicationData`. The tuple has several integer and float type 1-D tensor tuple elements. The first dimenion of this output_shape 2-D tensor is tensor type of tuple elements, `0` represents integer tensor, `1` represents float tensor. The second dimension of `output_shape` gives length of each tuple element.