XlaSparseCoreAdagradMomentum

public final class XlaSparseCoreAdagradMomentum

Public Methods

static XlaSparseCoreAdagradMomentum
create(Scope scope, Operand<Integer> indices, Operand<Float> gradient, Operand<Float> learningRate, Operand<Float> beta1, Operand<Float> epsilon, Operand<Float> accumulator, Operand<Float> momentum, Operand<Float> embeddingTable, Long featureWidth, Boolean useNesterov, Float beta2, Float exponent)
Factory method to create a class wrapping a new XlaSparseCoreAdagradMomentum operation.
Output<Float>
updatedAccumulator()
Output<Float>
updatedEmbeddingTable()
Output<Float>
updatedMomentum()

Inherited Methods

Public Methods

public static XlaSparseCoreAdagradMomentum create (Scope scope, Operand<Integer> indices, Operand<Float> gradient, Operand<Float> learningRate, Operand<Float> beta1, Operand<Float> epsilon, Operand<Float> accumulator, Operand<Float> momentum, Operand<Float> embeddingTable, Long featureWidth, Boolean useNesterov, Float beta2, Float exponent)

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

Parameters
scope current scope
Returns
  • a new instance of XlaSparseCoreAdagradMomentum

public Output<Float> updatedAccumulator ()

public Output<Float> updatedEmbeddingTable ()

public Output<Float> updatedMomentum ()