tf.raw_ops.XlaSparseDenseMatmulWithCsrInput

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.raw_ops.XlaSparseDenseMatmulWithCsrInput

tf.raw_ops.XlaSparseDenseMatmulWithCsrInput(
    row_pointers,
    sorted_sample_ids,
    sorted_token_ids,
    sorted_gains,
    embedding_table,
    num_minibatches_per_physical_sparse_core,
    input_size,
    quantization_config_low,
    quantization_config_high,
    quantization_config_num_buckets,
    table_name,
    num_sparsecores_per_device=-1,
    name=None
)

Args
row_pointers	A Tensor of type int32.
sorted_sample_ids	A Tensor of type int32.
sorted_token_ids	A Tensor of type int32.
sorted_gains	A Tensor of type float32.
embedding_table	A Tensor. Must be one of the following types: int32, float32.
num_minibatches_per_physical_sparse_core	A Tensor of type int32.
input_size	An int that is >= 0.
quantization_config_low	A float.
quantization_config_high	A float.
quantization_config_num_buckets	An int that is >= 0.
table_name	A string.
num_sparsecores_per_device	An optional int. Defaults to -1.
name	A name for the operation (optional).

Returns
A Tensor. Has the same type as embedding_table.