XlaSparseCoreFtrl

public final class XlaSparseCoreFtrl

Public Methods

static XlaSparseCoreFtrl
create(Scope scope, Operand<Float> embeddingTable, Operand<Float> accumulator, Operand<Float> linear, Operand<Float> learningRate, Operand<Integer> indices, Operand<Float> gradient, Operand<Float> beta, Operand<Float> learningRatePower, Operand<Float> l2RegularizationStrength, Long featureWidth, Boolean multiplyLinearByLearningRate, Float l1RegularizationStrength)
Factory method to create a class wrapping a new XlaSparseCoreFtrl operation.
Output<Float>
updatedAccumulator()
Output<Float>
updatedEmbeddingTable()
Output<Float>
updatedLinear()

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 XlaSparseCoreFtrl create (Scope scope, Operand<Float> embeddingTable, Operand<Float> accumulator, Operand<Float> linear, Operand<Float> learningRate, Operand<Integer> indices, Operand<Float> gradient, Operand<Float> beta, Operand<Float> learningRatePower, Operand<Float> l2RegularizationStrength, Long featureWidth, Boolean multiplyLinearByLearningRate, Float l1RegularizationStrength)

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

Parameters
scope current scope
Returns
  • a new instance of XlaSparseCoreFtrl

public Output<Float> updatedAccumulator ()

public Output<Float> updatedEmbeddingTable ()

public Output<Float> updatedLinear ()