RetrieveTPUEmbeddingFTRLParameters

public final class RetrieveTPUEmbeddingFTRLParameters

Retrieve FTRL 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 RetrieveTPUEmbeddingFTRLParameters.Options Optional attributes for RetrieveTPUEmbeddingFTRLParameters  

Public Methods

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

Inherited Methods

Public Methods

public Output<Float> accumulators ()

Parameter accumulators updated by the FTRL optimization algorithm.

public static RetrieveTPUEmbeddingFTRLParameters.Options config (String config)

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

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

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

public Output<Float> linears ()

Parameter linears updated by the FTRL optimization algorithm.

public Output<Float> parameters ()

Parameter parameters updated by the FTRL optimization algorithm.

public static RetrieveTPUEmbeddingFTRLParameters.Options tableId (Long tableId)

public static RetrieveTPUEmbeddingFTRLParameters.Options tableName (String tableName)