Warning: This API is deprecated and will be removed in a future version of TensorFlow after the replacement is stable.

LoadTPUEmbeddingFrequencyEstimatorParameters

public final class LoadTPUEmbeddingFrequencyEstimatorParameters

Load frequency estimator 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 LoadTPUEmbeddingFrequencyEstimatorParameters.Options Optional attributes for LoadTPUEmbeddingFrequencyEstimatorParameters

Public Methods

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

Inherited Methods

Public Methods

public static LoadTPUEmbeddingFrequencyEstimatorParameters.Options config (String config)

public static LoadTPUEmbeddingFrequencyEstimatorParameters create ( Scope scope, Operand <Float> parameters, Operand <Float> lastHitStep, Long numShards, Long shardId, Options... options)

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

Parameters
scope current scope
parameters Value of parameters used in the frequency estimator optimization algorithm.
lastHitStep Value of last_hit_step used in the frequency estimator optimization algorithm.
options carries optional attributes values
Returns
  • a new instance of LoadTPUEmbeddingFrequencyEstimatorParameters

public static LoadTPUEmbeddingFrequencyEstimatorParameters.Options tableId (Long tableId)

public static LoadTPUEmbeddingFrequencyEstimatorParameters.Options tableName (String tableName)