tf.raw_ops.LoadTPUEmbeddingADAMParameters

Load ADAM embedding parameters.

tf.raw_ops.LoadTPUEmbeddingADAMParameters(
    parameters,
    momenta,
    velocities,
    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 ADAM optimization algorithm.
momentaA Tensor of type float32. Value of momenta used in the ADAM optimization algorithm.
velocitiesA Tensor of type float32. Value of velocities used in the ADAM 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.