tf.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,
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. |
