XlaSparseDenseMatmulGradWithAdagradAndCsrInput

public final class XlaSparseDenseMatmulGradWithAdagradAndCsrInput

Nested Classes

class XlaSparseDenseMatmulGradWithAdagradAndCsrInput.Options Optional attributes for XlaSparseDenseMatmulGradWithAdagradAndCsrInput  

Public Methods

static XlaSparseDenseMatmulGradWithAdagradAndCsrInput.Options
clipWeightMax(Float clipWeightMax)
static XlaSparseDenseMatmulGradWithAdagradAndCsrInput.Options
clipWeightMin(Float clipWeightMin)
static XlaSparseDenseMatmulGradWithAdagradAndCsrInput
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<Integer> numMinibatchesPerPhysicalSparseCore, String tableName, Options... options)
Factory method to create a class wrapping a new XlaSparseDenseMatmulGradWithAdagradAndCsrInput operation.
Output<Float>
updatedAccumulator()
Output<Float>
updatedEmbeddingTable()

Inherited Methods

org.tensorflow.op.PrimitiveOp
final boolean
equals(Object obj)
final int
hashCode()
Operation
op()
Returns the underlying Operation
final String
toString()
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 XlaSparseDenseMatmulGradWithAdagradAndCsrInput.Options clipWeightMax (Float clipWeightMax)

public static XlaSparseDenseMatmulGradWithAdagradAndCsrInput.Options clipWeightMin (Float clipWeightMin)

public static XlaSparseDenseMatmulGradWithAdagradAndCsrInput 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<Integer> numMinibatchesPerPhysicalSparseCore, String tableName, Options... options)

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

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

public Output<Float> updatedAccumulator ()

public Output<Float> updatedEmbeddingTable ()