tf.raw_ops.XlaSparseCoreAdagradMomentum
tf.raw_ops.XlaSparseCoreAdagradMomentum(
indices,
gradient,
learning_rate,
beta_1,
epsilon,
accumulator,
momentum,
embedding_table,
feature_width,
use_nesterov,
beta_2,
exponent,
name=None
)
| Args |
|---|
indices | A Tensor of type int32. |
gradient | A Tensor of type float32. |
learning_rate | A Tensor of type float32. |
beta_1 | A Tensor of type float32. |
epsilon | A Tensor of type float32. |
accumulator | A Tensor of type float32. |
momentum | A Tensor of type float32. |
embedding_table | A Tensor of type float32. |
feature_width | An int. |
use_nesterov | A bool. |
beta_2 | A float. |
exponent | A float. |
name | A name for the operation (optional). |
| Returns |
|---|
A tuple of Tensor objects (updated_embedding_table, updated_accumulator, updated_momentum). |
updated_embedding_table | A Tensor of type float32. |
updated_accumulator | A Tensor of type float32. |
updated_momentum | A Tensor of type float32. |