A conditional accumulator for aggregating gradients.
The accumulator accepts gradients marked with local_step greater or equal to the most recent global_step known to the accumulator. The average can be extracted from the accumulator, provided sufficient gradients have been accumulated. Extracting the average automatically resets the aggregate to 0, and increments the global_step recorded by the accumulator. This is a resource version of ConditionalAccumulator that will work in TF2.0 with tf.cond version 2.
Nested Classes
class | ResourceConditionalAccumulator.Options |
Optional attributes for
ResourceConditionalAccumulator
|
Public Methods
Output <Object> |
asOutput
()
Returns the symbolic handle of a tensor.
|
static ResourceConditionalAccumulator.Options |
container
(String container)
|
static <T> ResourceConditionalAccumulator |
create
(
Scope
scope, Class<T> dtype,
Shape
shape,
Options...
options)
Factory method to create a class wrapping a new ResourceConditionalAccumulator operation.
|
Output <?> |
handle
()
The handle to the accumulator.
|
static ResourceConditionalAccumulator.Options |
reductionType
(String reductionType)
|
static ResourceConditionalAccumulator.Options |
sharedName
(String sharedName)
|
Inherited Methods
Public Methods
public Output <Object> asOutput ()
Returns the symbolic handle of a tensor.
Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.
public static ResourceConditionalAccumulator.Options container (String container)
Parameters
container | If non-empty, this accumulator is placed in the given container. Otherwise, a default container is used. |
---|
public static ResourceConditionalAccumulator create ( Scope scope, Class<T> dtype, Shape shape, Options... options)
Factory method to create a class wrapping a new ResourceConditionalAccumulator operation.
Parameters
scope | current scope |
---|---|
dtype | The type of the value being accumulated. |
shape | The shape of the values, can be [], in which case shape is unknown. |
options | carries optional attributes values |
Returns
- a new instance of ResourceConditionalAccumulator
public static ResourceConditionalAccumulator.Options sharedName (String sharedName)
Parameters
sharedName | If non-empty, this accumulator will be shared under the given name across multiple sessions. |
---|