SparseAccumulatorApplyGradient

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
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.

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