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,
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 of type 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 .
|
name
|
A name for the operation (optional).
|
Returns |
A Tensor of type float32 .
|