tf.raw_ops.RetrieveTPUEmbeddingStochasticGradientDescentParameters
Retrieve SGD embedding parameters.
View aliases
Compat aliases for migration
See
Migration guide for
more details.
tf.compat.v1.raw_ops.RetrieveTPUEmbeddingStochasticGradientDescentParameters
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.
Args
num_shards
An int
.
shard_id
An int
.
table_id
An optional int
. Defaults to -1
.
table_name
An optional string
. Defaults to ""
.
config
An optional string
. Defaults to ""
.
name
A name for the operation (optional).
Returns
A Tensor
of type float32
.
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License , and code samples are licensed under the Apache 2.0 License . For details, see the Google Developers Site Policies . Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license .
Last updated 2024-01-23 UTC.
[null,null,["Last updated 2024-01-23 UTC."],[],[]]