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

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 ()