Applies a sparse gradient to a given accumulator.
tf.raw_ops.SparseAccumulatorApplyGradient(
handle,
local_step,
gradient_indices,
gradient_values,
gradient_shape,
has_known_shape,
name=None
)
Does not add if local_step is smaller than the accumulator's global_step.
Args |
|---|
handle
Tensor of type mutable string. The handle to a accumulator.
local_step
Tensor of type int64.
The local_step value at which the sparse gradient was computed.
gradient_indices
Tensor of type int64.
Indices of the sparse gradient to be accumulated. Must be a
vector.
gradient_values
Tensor. Must be one of the following types: float32, float64, int32, uint8, int16, int8, complex64, int64, qint8, quint8, qint32, bfloat16, qint16, quint16, uint16, complex128, half, uint32, uint64.
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.
gradient_shape
Tensor of type int64.
Shape of the sparse gradient to be accumulated.
has_known_shape
bool.
Boolean indicating whether gradient_shape is unknown, in which
case the input is ignored during validation.
name
Returns | |
|---|---|
| The created Operation. |