tf.raw_ops.XlaSparseDenseMatmulGradWithAdamAndCsrInput

row_pointersA Tensor of type int32.
sorted_sample_idsA Tensor of type int32.
sorted_token_idsA Tensor of type int32.
sorted_gainsA Tensor of type float32.
activation_gradientsA Tensor of type float32.
learning_rateA Tensor of type float32.
embedding_tableA Tensor of type float32.
momentaA Tensor of type float32.
velocityA Tensor of type float32.
num_minibatches_per_physical_sparse_coreA Tensor of type int32.
use_sum_inside_sqrtA bool.
beta1A float.
beta2A float.
epsilonA float.
table_nameA string.
clip_weight_minAn optional float. Defaults to float('-inf').
clip_weight_maxAn optional float. Defaults to float('inf').
nameA name for the operation (optional).

A tuple of Tensor objects (updated_embedding_table, updated_momenta, updated_velocity).
updated_embedding_tableA Tensor of type float32.
updated_momentaA Tensor of type float32.
updated_velocityA Tensor of type float32.