tf.raw_ops.XlaSparseCoreAdagrad
tf.raw_ops.XlaSparseCoreAdagrad(
indices,
gradient,
learning_rate,
accumulator,
embedding_table,
feature_width,
name=None
)
| Args |
|---|
indices | A Tensor of type int32. |
gradient | A Tensor of type float32. |
learning_rate | A Tensor of type float32. |
accumulator | A Tensor of type float32. |
embedding_table | A Tensor of type float32. |
feature_width | An int. |
name | A name for the operation (optional). |
| Returns |
|---|
A tuple of Tensor objects (updated_embedding_table, updated_accumulator). |
updated_embedding_table | A Tensor of type float32. |
updated_accumulator | A Tensor of type float32. |
