Retrieve Adagrad Momentum embedding parameters.
View aliases
Compat aliases for migration
See Migration guide for more details.
tf.compat.v1.raw_ops.RetrieveTPUEmbeddingAdagradMomentumParameters
tf.raw_ops.RetrieveTPUEmbeddingAdagradMomentumParameters(
num_shards,
shard_id,
table_id=-1,
table_name='',
config='',
name=None
)
An op that retrieves optimization parameters from embedding to host memory. Must be preceded by a ConfigureTPUEmbeddingHost op that sets up the correct embedding table configuration. For example, this op is used to retrieve updated parameters before saving a checkpoint.