RetrieveTPUEmbeddingRMSPropParameters

public final class RetrieveTPUEmbeddingRMSPropParameters

Retrieve RMSProp 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 RetrieveTPUEmbeddingRMSPropParameters.Options Optional attributes for RetrieveTPUEmbeddingRMSPropParameters  

Public Methods

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

Inherited Methods

Public Methods

public static RetrieveTPUEmbeddingRMSPropParameters.Options config (String config)

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

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

Parameters
scope current scope
options carries optional attributes values
Returns
  • a new instance of RetrieveTPUEmbeddingRMSPropParameters

public Output<Float> mom ()

Parameter mom updated by the RMSProp optimization algorithm.

public Output<Float> ms ()

Parameter ms updated by the RMSProp optimization algorithm.

public Output<Float> parameters ()

Parameter parameters updated by the RMSProp optimization algorithm.

public static RetrieveTPUEmbeddingRMSPropParameters.Options tableId (Long tableId)

public static RetrieveTPUEmbeddingRMSPropParameters.Options tableName (String tableName)