RetrieveTPUEmbeddingADAMParameters

public final class RetrieveTPUEmbeddingADAMParameters

Retrieve ADAM embedding parameters.

An op that retrieves optimization parameters from embedding to host memory. Must be preceded by a ConfigureTPUEmbeddingHost op that sets up the correct embedding table configuration. For example, this op is used to retrieve updated parameters before saving a checkpoint.

Nested Classes

class

RetrieveTPUEmbeddingADAMParameters.Options

Optional attributes for RetrieveTPUEmbeddingADAMParameters

Public Methods

static RetrieveTPUEmbeddingADAMParameters.Options	config (String config)
static RetrieveTPUEmbeddingADAMParameters	create ( Scope scope, Long numShards, Long shardId, Options... options) Factory method to create a class wrapping a new RetrieveTPUEmbeddingADAMParameters operation.
Output <Float>	momenta () Parameter momenta updated by the ADAM optimization algorithm.
Output <Float>	parameters () Parameter parameters updated by the ADAM optimization algorithm.
static RetrieveTPUEmbeddingADAMParameters.Options	tableId (Long tableId)
static RetrieveTPUEmbeddingADAMParameters.Options	tableName (String tableName)
Output <Float>	velocities () Parameter velocities updated by the ADAM optimization algorithm.

Inherited Methods

From class org.tensorflow.op.PrimitiveOp

final boolean	equals (Object obj)
final int	hashCode ()
Operation	op () Returns the underlying Operation
final String	toString ()

From class java.lang.Object

boolean	equals (Object arg0)
final Class<?>	getClass ()
int	hashCode ()
final void	notify ()
final void	notifyAll ()
String	toString ()
final void	wait (long arg0, int arg1)
final void	wait (long arg0)
final void	wait ()

Public Methods

public static RetrieveTPUEmbeddingADAMParameters.Options config (String config)

public static RetrieveTPUEmbeddingADAMParameters create ( Scope scope, Long numShards, Long shardId, Options... options)

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

Parameters

scope	current scope
options	carries optional attributes values

Returns

• a new instance of RetrieveTPUEmbeddingADAMParameters

public Output <Float> momenta ()

Parameter momenta updated by the ADAM optimization algorithm.

public Output <Float> parameters ()

Parameter parameters updated by the ADAM optimization algorithm.

public static RetrieveTPUEmbeddingADAMParameters.Options tableId (Long tableId)

public static RetrieveTPUEmbeddingADAMParameters.Options tableName (String tableName)

public Output <Float> velocities ()

Parameter velocities updated by the ADAM optimization algorithm.