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 .
|