tf.raw_ops.XlaSparseCoreAdam

tf.raw_ops.XlaSparseCoreAdam(
    embedding_table,
    indices,
    gradient,
    learning_rate,
    momentum,
    velocity,
    beta_1,
    beta_2,
    epsilon,
    feature_width,
    use_sum_inside_sqrt,
    name=None
)

embedding_table A Tensor of type float32.
indices A Tensor of type int32.
gradient A Tensor of type float32.
learning_rate A Tensor of type float32.
momentum A Tensor of type float32.
velocity A Tensor of type float32.
beta_1 A Tensor of type float32.
beta_2 A Tensor of type float32.
epsilon A Tensor of type float32.
feature_width An int.
use_sum_inside_sqrt A bool.
name A name for the operation (optional).

A tuple of Tensor objects (updated_embedding_table, updated_velocity, updated_momentum).
updated_embedding_table A Tensor of type float32.
updated_velocity A Tensor of type float32.
updated_momentum A Tensor of type float32.