LoadTPUEmbeddingAdadeltaParameters

public final class LoadTPUEmbeddingAdadeltaParameters

Load Adadelta embedding parameters.

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 LoadTPUEmbeddingAdadeltaParameters.Options Optional attributes for LoadTPUEmbeddingAdadeltaParameters  

Public Methods

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

Inherited Methods

Public Methods

public static LoadTPUEmbeddingAdadeltaParameters.Options config (String config)

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

Factory method to create a class wrapping a new LoadTPUEmbeddingAdadeltaParameters 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.
options carries optional attributes values
Returns
  • a new instance of LoadTPUEmbeddingAdadeltaParameters

public static LoadTPUEmbeddingAdadeltaParameters.Options tableId (Long tableId)

public static LoadTPUEmbeddingAdadeltaParameters.Options tableName (String tableName)