tf.raw_ops.LoadTPUEmbeddingRMSPropParameters

Load RMSProp embedding parameters.

tf.raw_ops.LoadTPUEmbeddingRMSPropParameters(
    parameters,
    ms,
    mom,
    num_shards,
    shard_id,
    table_id=-1,
    table_name='',
    config='',
    name=None
)

An op that loads optimization parameters into HBM for embedding. Must be preceded by a ConfigureTPUEmbeddingHost op that sets up the correct embedding table configuration. For example, this op is used to install parameters that are loaded from a checkpoint before a training loop is executed.

parametersA Tensor of type float32. Value of parameters used in the RMSProp optimization algorithm.
msA Tensor of type float32. Value of ms used in the RMSProp optimization algorithm.
momA Tensor of type float32. Value of mom used in the RMSProp optimization algorithm.
num_shardsAn int.
shard_idAn int.
table_idAn optional int. Defaults to -1.
table_nameAn optional string. Defaults to "".
configAn optional string. Defaults to "".
nameA name for the operation (optional).

The created Operation.