RetrieveTPUEmbeddingADAMParameters

public final class RetrieveTPUEmbeddingADAMParameters

Retrieve ADAM 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 RetrieveTPUEmbeddingADAMParameters.Options Optional attributes for RetrieveTPUEmbeddingADAMParameters  

Public Methods

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

Inherited Methods

Public Methods

public static RetrieveTPUEmbeddingADAMParameters.Options config (String config)

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

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

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

public Output<Float> momenta ()

Parameter momenta updated by the ADAM optimization algorithm.

public Output<Float> parameters ()

Parameter parameters updated by the ADAM optimization algorithm.

public static RetrieveTPUEmbeddingADAMParameters.Options tableId (Long tableId)

public static RetrieveTPUEmbeddingADAMParameters.Options tableName (String tableName)

public Output<Float> velocities ()

Parameter velocities updated by the ADAM optimization algorithm.