tf.raw_ops.LoadTPUEmbeddingFTRLParameters

Load FTRL embedding parameters.

tf.raw_ops.LoadTPUEmbeddingFTRLParameters(
    parameters,
    accumulators,
    linears,
    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 FTRL optimization algorithm.
accumulatorsA Tensor of type float32. Value of accumulators used in the FTRL optimization algorithm.
linearsA Tensor of type float32. Value of linears used in the FTRL 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.