tf.raw_ops.XlaSparseDenseMatmulWithCsrInput

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).

A Tensor of type float32.