A conditional accumulator for aggregating sparse 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.
Nested Classes
class | SparseConditionalAccumulator.Options | Optional attributes for SparseConditionalAccumulator
|
Constants
String | OP_NAME | The name of this op, as known by TensorFlow core engine |
Public Methods
Output<TString> |
asOutput()
Returns the symbolic handle of the tensor.
|
static SparseConditionalAccumulator.Options |
container(String container)
|
static <T extends TType> SparseConditionalAccumulator |
create(Scope scope, Class<T> dtype, Shape shape, Options... options)
Factory method to create a class wrapping a new SparseConditionalAccumulator operation.
|
Output<TString> |
handle()
The handle to the accumulator.
|
static SparseConditionalAccumulator.Options |
reductionType(String reductionType)
|
static SparseConditionalAccumulator.Options |
sharedName(String sharedName)
|
Inherited Methods
Constants
public static final String OP_NAME
The name of this op, as known by TensorFlow core engine
Public Methods
public Output<TString> asOutput ()
Returns the symbolic handle of the 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 SparseConditionalAccumulator.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 SparseConditionalAccumulator create (Scope scope, Class<T> dtype, Shape shape, Options... options)
Factory method to create a class wrapping a new SparseConditionalAccumulator operation.
Parameters
scope | current scope |
---|---|
dtype | The type of the value being accumulated. |
shape | The shape of the values. |
options | carries optional attributes values |
Returns
- a new instance of SparseConditionalAccumulator
public static SparseConditionalAccumulator.Options sharedName (String sharedName)
Parameters
sharedName | If non-empty, this accumulator will be shared under the given name across multiple sessions. |
---|