XlaSparseDenseMatmulGradWithAdamAndCsrInput

public final class XlaSparseDenseMatmulGradWithAdamAndCsrInput

Nested Classes

class

XlaSparseDenseMatmulGradWithAdamAndCsrInput.Options

Optional attributes for XlaSparseDenseMatmulGradWithAdamAndCsrInput

Public Methods

static XlaSparseDenseMatmulGradWithAdamAndCsrInput.Options	clipWeightMax(Float clipWeightMax)
static XlaSparseDenseMatmulGradWithAdamAndCsrInput.Options	clipWeightMin(Float clipWeightMin)
static XlaSparseDenseMatmulGradWithAdamAndCsrInput	create(Scope scope, Operand<Integer> rowPointers, Operand<Integer> sortedSampleIds, Operand<Integer> sortedTokenIds, Operand<Float> sortedGains, Operand<Float> activationGradients, Operand<Float> learningRate, Operand<Float> embeddingTable, Operand<Float> momenta, Operand<Float> velocity, Operand<Integer> numMinibatchesPerPhysicalSparseCore, Boolean useSumInsideSqrt, Float beta1, Float beta2, Float epsilon, String tableName, Options... options) Factory method to create a class wrapping a new XlaSparseDenseMatmulGradWithAdamAndCsrInput operation.
Output<Float>	updatedEmbeddingTable()
Output<Float>	updatedMomenta()
Output<Float>	updatedVelocity()

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 XlaSparseDenseMatmulGradWithAdamAndCsrInput.Options clipWeightMax (Float clipWeightMax)

public static XlaSparseDenseMatmulGradWithAdamAndCsrInput.Options clipWeightMin (Float clipWeightMin)

public static XlaSparseDenseMatmulGradWithAdamAndCsrInput create (Scope scope, Operand<Integer> rowPointers, Operand<Integer> sortedSampleIds, Operand<Integer> sortedTokenIds, Operand<Float> sortedGains, Operand<Float> activationGradients, Operand<Float> learningRate, Operand<Float> embeddingTable, Operand<Float> momenta, Operand<Float> velocity, Operand<Integer> numMinibatchesPerPhysicalSparseCore, Boolean useSumInsideSqrt, Float beta1, Float beta2, Float epsilon, String tableName, Options... options)

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

Parameters

scope	current scope
options	carries optional attributes values

Returns

• a new instance of XlaSparseDenseMatmulGradWithAdamAndCsrInput

public Output<Float> updatedEmbeddingTable ()

public Output<Float> updatedMomenta ()

public Output<Float> updatedVelocity ()