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> | |
Output<Float> | |
Output<Float> |
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