XlaSparseCoreAdam

public final class XlaSparseCoreAdam

Public Methods

static XlaSparseCoreAdam	create(Scope scope, Operand<Float> embeddingTable, Operand<Integer> indices, Operand<Float> gradient, Operand<Float> learningRate, Operand<Float> momentum, Operand<Float> velocity, Operand<Float> beta1, Operand<Float> beta2, Operand<Float> epsilon, Long featureWidth, Boolean useSumInsideSqrt) Factory method to create a class wrapping a new XlaSparseCoreAdam operation.
Output<Float>	updatedEmbeddingTable()
Output<Float>	updatedMomentum()
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 XlaSparseCoreAdam create (Scope scope, Operand<Float> embeddingTable, Operand<Integer> indices, Operand<Float> gradient, Operand<Float> learningRate, Operand<Float> momentum, Operand<Float> velocity, Operand<Float> beta1, Operand<Float> beta2, Operand<Float> epsilon, Long featureWidth, Boolean useSumInsideSqrt)

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

Parameters

scope	current scope

Returns

• a new instance of XlaSparseCoreAdam

public Output<Float> updatedEmbeddingTable ()

public Output<Float> updatedMomentum ()

public Output<Float> updatedVelocity ()