XlaSparseCoreAdagrad

public final class XlaSparseCoreAdagrad

Public Methods

static XlaSparseCoreAdagrad
create(Scope scope, Operand<Integer> indices, Operand<Float> gradient, Operand<Float> learningRate, Operand<Float> accumulator, Operand<Float> embeddingTable, Long featureWidth)
Factory method to create a class wrapping a new XlaSparseCoreAdagrad operation.
Output<Float>
Output<Float>

Inherited Methods

Public Methods

public static XlaSparseCoreAdagrad create (Scope scope, Operand<Integer> indices, Operand<Float> gradient, Operand<Float> learningRate, Operand<Float> accumulator, Operand<Float> embeddingTable, Long featureWidth)

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

Parameters
scope current scope
Returns
  • a new instance of XlaSparseCoreAdagrad

public Output<Float> updatedAccumulator ()

public Output<Float> updatedEmbeddingTable ()