Retrieve SGD embedding parameters.
tf.raw_ops.RetrieveTPUEmbeddingStochasticGradientDescentParameters(
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.
Returns | |
---|---|
A Tensor of type float32 .
|