public final class
SparseAccumulatorApplyGradient
Applies a sparse gradient to a given accumulator.
Does not add if local_step is smaller than the accumulator's global_step.
Constants
String | OP_NAME | The name of this op, as known by TensorFlow core engine |
Public Methods
static SparseAccumulatorApplyGradient |
Inherited Methods
Constants
public static final String OP_NAME
The name of this op, as known by TensorFlow core engine
Constant Value:
"SparseAccumulatorApplyGradient"
Public Methods
public static SparseAccumulatorApplyGradient create (Scope scope, Operand<TString> handle, Operand<TInt64> localStep, Operand<TInt64> gradientIndices, Operand<? extends TType> gradientValues, Operand<TInt64> gradientShape, Boolean hasKnownShape)
Factory method to create a class wrapping a new SparseAccumulatorApplyGradient operation.
Parameters
scope | current scope |
---|---|
handle | The handle to a accumulator. |
localStep | The local_step value at which the sparse gradient was computed. |
gradientIndices | Indices of the sparse gradient to be accumulated. Must be a vector. |
gradientValues | Values are the non-zero slices of the gradient, and must have the same first dimension as indices, i.e., the nnz represented by indices and values must be consistent. |
gradientShape | Shape of the sparse gradient to be accumulated. |
hasKnownShape | Boolean indicating whether gradient_shape is unknown, in which case the input is ignored during validation. |
Returns
- a new instance of SparseAccumulatorApplyGradient