tensorflow:: ops:: ConditionalAccumulator
#include <data_flow_ops.h>
A conditional accumulator for aggregating gradients.
Summary
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.
Args:
- scope: A Scope object
- dtype: The type of the value being accumulated.
- shape: The shape of the values, can be [], in which case shape is unknown.
Optional attributes (see Attrs
):
- container: If non-empty, this accumulator is placed in the given container. Otherwise, a default container is used.
- shared_name: If non-empty, this accumulator will be shared under the given name across multiple sessions.
Returns:
Output
: The handle to the accumulator.
Constructors and Destructors |
|
---|---|
ConditionalAccumulator(const ::tensorflow::Scope & scope, DataType dtype, PartialTensorShape shape)
|
|
ConditionalAccumulator(const ::tensorflow::Scope & scope, DataType dtype, PartialTensorShape shape, const ConditionalAccumulator::Attrs & attrs)
|
Public attributes |
|
---|---|
handle
|
|
operation
|
Public functions |
|
---|---|
node() const
|
::tensorflow::Node *
|
operator::tensorflow::Input() const
|
|
operator::tensorflow::Output() const
|
|
Public static functions |
|
---|---|
Container(StringPiece x)
|
|
ReductionType(StringPiece x)
|
|
SharedName(StringPiece x)
|
Structs |
|
---|---|
tensorflow:: |
Optional attribute setters for ConditionalAccumulator. |
Public attributes
handle
::tensorflow::Output handle
operation
Operation operation
Public functions
ConditionalAccumulator
ConditionalAccumulator( const ::tensorflow::Scope & scope, DataType dtype, PartialTensorShape shape )
ConditionalAccumulator
ConditionalAccumulator( const ::tensorflow::Scope & scope, DataType dtype, PartialTensorShape shape, const ConditionalAccumulator::Attrs & attrs )
node
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const
Public static functions
Container
Attrs Container( StringPiece x )
ReductionType
Attrs ReductionType( StringPiece x )
SharedName
Attrs SharedName( StringPiece x )