Class LoadTPUEmbeddingProximalAdagradParameters

java.lang.Object

org.tensorflow.op.RawOp

org.tensorflow.op.tpu.LoadTPUEmbeddingProximalAdagradParameters

All Implemented Interfaces:

Op

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

Load proximal Adagrad 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	LoadTPUEmbeddingProximalAdagradParameters.Inputs
static class	LoadTPUEmbeddingProximalAdagradParameters.Options	Optional attributes for LoadTPUEmbeddingProximalAdagradParameters

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
LoadTPUEmbeddingProximalAdagradParameters(Operation operation)

Method Summary

Modifier and Type	Method	Description
static LoadTPUEmbeddingProximalAdagradParameters.Options	config(String config)	Sets the config option.
static LoadTPUEmbeddingProximalAdagradParameters	create(Scope scope, Operand<TFloat32> parameters, Operand<TFloat32> accumulators, Long numShards, Long shardId, LoadTPUEmbeddingProximalAdagradParameters.Options... options)	Factory method to create a class wrapping a new LoadTPUEmbeddingProximalAdagradParameters operation.
static LoadTPUEmbeddingProximalAdagradParameters.Options	tableId(Long tableId)	Sets the tableId option.
static LoadTPUEmbeddingProximalAdagradParameters.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

LoadTPUEmbeddingProximalAdagradParameters

public LoadTPUEmbeddingProximalAdagradParameters(Operation operation)

Method Details

create

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

Factory method to create a class wrapping a new LoadTPUEmbeddingProximalAdagradParameters 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.

numShards - The value of the numShards attribute

shardId - The value of the shardId attribute

options - carries optional attribute values

Returns:

a new instance of LoadTPUEmbeddingProximalAdagradParameters

tableId

public static LoadTPUEmbeddingProximalAdagradParameters.Options tableId(Long tableId)

Sets the tableId option.

Parameters:

tableId - the tableId option

Returns:

this Options instance.

tableName

public static LoadTPUEmbeddingProximalAdagradParameters.Options tableName(String tableName)

Sets the tableName option.

Parameters:

tableName - the tableName option

Returns:

this Options instance.

config

public static LoadTPUEmbeddingProximalAdagradParameters.Options config(String config)

Sets the config option.

Parameters:

config - the config option

Returns:

this Options instance.