Class LoadTPUEmbeddingCenteredRMSPropParameters

java.lang.Object

org.tensorflow.op.RawOp

org.tensorflow.op.tpu.LoadTPUEmbeddingCenteredRMSPropParameters

All Implemented Interfaces:

Op

@Operator(group="tpu")
public final class LoadTPUEmbeddingCenteredRMSPropParameters
extends RawOp

Load centered RMSProp 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 Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	LoadTPUEmbeddingCenteredRMSPropParameters.Inputs
static class	LoadTPUEmbeddingCenteredRMSPropParameters.Options	Optional attributes for LoadTPUEmbeddingCenteredRMSPropParameters

Field Summary

Fields

Modifier and Type	Field	Description
static final String	OP_NAME	The name of this op, as known by TensorFlow core engine

Fields inherited from class RawOp

operation

Constructor Summary

Constructors

Constructor	Description
LoadTPUEmbeddingCenteredRMSPropParameters(Operation operation)

Method Summary

Modifier and Type	Method	Description
static LoadTPUEmbeddingCenteredRMSPropParameters.Options	config(String config)	Sets the config option.
static LoadTPUEmbeddingCenteredRMSPropParameters	create(Scope scope, Operand<TFloat32> parameters, Operand<TFloat32> ms, Operand<TFloat32> mom, Operand<TFloat32> mg, Long numShards, Long shardId, LoadTPUEmbeddingCenteredRMSPropParameters.Options... options)	Factory method to create a class wrapping a new LoadTPUEmbeddingCenteredRMSPropParameters operation.
static LoadTPUEmbeddingCenteredRMSPropParameters.Options	tableId(Long tableId)	Sets the tableId option.
static LoadTPUEmbeddingCenteredRMSPropParameters.Options	tableName(String tableName)	Sets the tableName option.

Methods inherited from class RawOp

equals, hashCode, op, toString

Methods inherited from class Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface Op

env

Field Details

OP_NAME

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

See Also:

• Constant Field Values

Constructor Details

LoadTPUEmbeddingCenteredRMSPropParameters

public LoadTPUEmbeddingCenteredRMSPropParameters(Operation operation)

Method Details

create

@Endpoint(describeByClass=true)
public static LoadTPUEmbeddingCenteredRMSPropParameters create(Scope scope,
 Operand<TFloat32> parameters,
 Operand<TFloat32> ms,
 Operand<TFloat32> mom,
 Operand<TFloat32> mg,
 Long numShards,
 Long shardId,
 LoadTPUEmbeddingCenteredRMSPropParameters.Options... options)

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

Parameters:

scope - current scope

parameters - Value of parameters used in the centered RMSProp optimization algorithm.

ms - Value of ms used in the centered RMSProp optimization algorithm.

mom - Value of mom used in the centered RMSProp optimization algorithm.

mg - Value of mg used in the centered RMSProp optimization algorithm.

numShards - The value of the numShards attribute

shardId - The value of the shardId attribute

options - carries optional attribute values

Returns:

a new instance of LoadTPUEmbeddingCenteredRMSPropParameters

tableId

public static LoadTPUEmbeddingCenteredRMSPropParameters.Options tableId(Long tableId)

Sets the tableId option.

Parameters:

tableId - the tableId option

Returns:

this Options instance.

tableName

public static LoadTPUEmbeddingCenteredRMSPropParameters.Options tableName(String tableName)

Sets the tableName option.

Parameters:

tableName - the tableName option

Returns:

this Options instance.

config

public static LoadTPUEmbeddingCenteredRMSPropParameters.Options config(String config)

Sets the config option.

Parameters:

config - the config option

Returns:

this Options instance.