XlaSparseDenseMatmulGradWithAdagradMomentumAndCsrInput

public final class XlaSparseDenseMatmulGradWithAdagradMomentumAndCsrInput

Nested Classes

class XlaSparseDenseMatmulGradWithAdagradMomentumAndCsrInput.Options Optional attributes for XlaSparseDenseMatmulGradWithAdagradMomentumAndCsrInput  

Public Methods

static XlaSparseDenseMatmulGradWithAdagradMomentumAndCsrInput.Options
clipWeightMax(Float clipWeightMax)
static XlaSparseDenseMatmulGradWithAdagradMomentumAndCsrInput.Options
clipWeightMin(Float clipWeightMin)
static XlaSparseDenseMatmulGradWithAdagradMomentumAndCsrInput
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> accumulator, Operand<Float> momenta, Operand<Integer> numMinibatchesPerPhysicalSparseCore, Boolean useNesterov, Float exponent, Float beta1, Float beta2, Float epsilon, String tableName, Options... options)
Factory method to create a class wrapping a new XlaSparseDenseMatmulGradWithAdagradMomentumAndCsrInput operation.
Output<Float>
updatedAccumulator()
Output<Float>
updatedEmbeddingTable()
Output<Float>
updatedMomenta()

Inherited Methods

Public Methods

public static XlaSparseDenseMatmulGradWithAdagradMomentumAndCsrInput.Options clipWeightMax (Float clipWeightMax)

public static XlaSparseDenseMatmulGradWithAdagradMomentumAndCsrInput.Options clipWeightMin (Float clipWeightMin)

public static XlaSparseDenseMatmulGradWithAdagradMomentumAndCsrInput 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> accumulator, Operand<Float> momenta, Operand<Integer> numMinibatchesPerPhysicalSparseCore, Boolean useNesterov, Float exponent, Float beta1, Float beta2, Float epsilon, String tableName, Options... options)

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

Parameters
scope current scope
options carries optional attributes values
Returns
  • a new instance of XlaSparseDenseMatmulGradWithAdagradMomentumAndCsrInput

public Output<Float> updatedAccumulator ()

public Output<Float> updatedEmbeddingTable ()

public Output<Float> updatedMomenta ()