tf.raw_ops.XlaSparseDenseMatmulGradWithSgdAndCsrInput

tf.raw_ops.XlaSparseDenseMatmulGradWithSgdAndCsrInput(
    row_pointers,
    sorted_sample_ids,
    sorted_token_ids,
    sorted_gains,
    activation_gradients,
    learning_rate,
    embedding_table,
    num_minibatches_per_physical_sparse_core,
    table_name,
    clip_weight_min=float('-inf'),
    clip_weight_max=float('inf'),
    name=None
)

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.
num_minibatches_per_physical_sparse_coreA Tensor of type int32.
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 Tensor of type float32.