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

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 ()