RetrieveTPUEmbeddingMomentumParameters

public final class RetrieveTPUEmbeddingMomentumParameters

Retrieve Momentum embedding parameters.

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.

Nested Classes

class RetrieveTPUEmbeddingMomentumParameters.Options Optional attributes for RetrieveTPUEmbeddingMomentumParameters  

Public Methods

static RetrieveTPUEmbeddingMomentumParameters.Options
config(String config)
static RetrieveTPUEmbeddingMomentumParameters
create(Scope scope, Long numShards, Long shardId, Options... options)
Factory method to create a class wrapping a new RetrieveTPUEmbeddingMomentumParameters operation.
Output<Float>
momenta()
Parameter momenta updated by the Momentum optimization algorithm.
Output<Float>
parameters()
Parameter parameters updated by the Momentum optimization algorithm.
static RetrieveTPUEmbeddingMomentumParameters.Options
tableId(Long tableId)
static RetrieveTPUEmbeddingMomentumParameters.Options
tableName(String tableName)

Inherited Methods

Public Methods

public static RetrieveTPUEmbeddingMomentumParameters.Options config (String config)

public static RetrieveTPUEmbeddingMomentumParameters create (Scope scope, Long numShards, Long shardId, Options... options)

Factory method to create a class wrapping a new RetrieveTPUEmbeddingMomentumParameters operation.

Parameters
scope current scope
options carries optional attributes values
Returns
  • a new instance of RetrieveTPUEmbeddingMomentumParameters

public Output<Float> momenta ()

Parameter momenta updated by the Momentum optimization algorithm.

public Output<Float> parameters ()

Parameter parameters updated by the Momentum optimization algorithm.

public static RetrieveTPUEmbeddingMomentumParameters.Options tableId (Long tableId)

public static RetrieveTPUEmbeddingMomentumParameters.Options tableName (String tableName)