LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug

public final class LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug

Load proximal Adagrad embedding parameters with debug support.

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.

Nested Classes

class LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.Options Optional attributes for LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug  

Public Methods

static LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.Options
config(String config)
static LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug
create(Scope scope, Operand<Float> parameters, Operand<Float> accumulators, Operand<Float> gradientAccumulators, Long numShards, Long shardId, Options... options)
Factory method to create a class wrapping a new LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug operation.
static LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.Options
tableId(Long tableId)
static LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.Options
tableName(String tableName)

Inherited Methods

Public Methods

public static LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug create (Scope scope, Operand<Float> parameters, Operand<Float> accumulators, Operand<Float> gradientAccumulators, Long numShards, Long shardId, Options... options)

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

Parameters
scope current scope
parameters Value of parameters used in the proximal Adagrad optimization algorithm.
accumulators Value of accumulators used in the proximal Adagrad optimization algorithm.
gradientAccumulators Value of gradient_accumulators used in the proximal Adagrad optimization algorithm.
options carries optional attributes values
Returns
  • a new instance of LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug

public static LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.Options tableName (String tableName)