org.tensorflow.op.core

Class RetrieveTPUEmbeddingMomentumParametersGradAccumDebug

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.RetrieveTPUEmbeddingMomentumParametersGradAccumDebug

All Implemented Interfaces:

Op

public final class RetrieveTPUEmbeddingMomentumParametersGradAccumDebug
extends PrimitiveOp

Retrieve Momentum embedding parameters with debug support.

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 Class Summary

Nested Classes

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

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug	create(Scope scope, Long numShards, Long shardId, RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options... options) Factory method to create a class wrapping a new RetrieveTPUEmbeddingMomentumParametersGradAccumDebug operation.
Output<Float>	gradientAccumulators() Parameter gradient_accumulators updated by the Momentum optimization algorithm.
Output<Float>	momenta() Parameter momenta updated by the Momentum optimization algorithm.
Output<Float>	parameters() Parameter parameters updated by the Momentum optimization algorithm.
static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options	tableId(Long tableId)
static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options	tableName(String tableName)

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, op, toString

Methods inherited from class java.lang.Object

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

Method Detail

create

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

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

Parameters:

scope - current scope

numShards -

shardId -

options - carries optional attributes values

Returns:

a new instance of RetrieveTPUEmbeddingMomentumParametersGradAccumDebug

tableId

public static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options tableId(Long tableId)

Parameters:

tableId -

tableName

public static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options tableName(String tableName)

Parameters:

tableName -

parameters

public Output<Float> parameters()

Parameter parameters updated by the Momentum optimization algorithm.

momenta

public Output<Float> momenta()

Parameter momenta updated by the Momentum optimization algorithm.

gradientAccumulators

public Output<Float> gradientAccumulators()

Parameter gradient_accumulators updated by the Momentum optimization algorithm.