tf.raw_ops.LoadTPUEmbeddingMDLAdagradLightParameters

Load MDL Adagrad Light embedding parameters.

tf.raw_ops.LoadTPUEmbeddingMDLAdagradLightParameters(
    parameters,
    accumulators,
    weights,
    benefits,
    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 MDL Adagrad Light optimization algorithm.
accumulatorsA Tensor of type float32. Value of accumulators used in the MDL Adagrad Light optimization algorithm.
weightsA Tensor of type float32. Value of weights used in the MDL Adagrad Light optimization algorithm.
benefitsA Tensor of type float32. Value of benefits used in the MDL Adagrad Light 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.