LoadTPUEmbeddingAdadeltaParametersGradAccumDebug

public final class LoadTPUEmbeddingAdadeltaParametersGradAccumDebug

Load Adadelta 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 LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.Options Optional attributes for LoadTPUEmbeddingAdadeltaParametersGradAccumDebug  

Public Methods

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

Inherited Methods

Public Methods

public static LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.Options config (String config)

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

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

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

public static LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.Options tableId (Long tableId)

public static LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.Options tableName (String tableName)