public final class ResourceAccumulatorApplyGradient extends PrimitiveOp
Does not add if local_step is lesser than the accumulator's global_step.
operation
Modifier and Type | Method and Description |
---|---|
static <T> ResourceAccumulatorApplyGradient |
create(Scope scope,
Operand<?> handle,
Operand<Long> localStep,
Operand<T> gradient)
Factory method to create a class wrapping a new ResourceAccumulatorApplyGradient operation.
|
equals, hashCode, op, toString
public static <T> ResourceAccumulatorApplyGradient create(Scope scope, Operand<?> handle, Operand<Long> localStep, Operand<T> gradient)
scope
- current scopehandle
- The handle to a accumulator.localStep
- The local_step value at which the gradient was computed.gradient
- A tensor of the gradient to be accumulated.Copyright © 2022. All rights reserved.