#include <data_flow_ops.h>
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.
Arguments:
Optional attributes (see Attrs
):
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 |
Public functions | |
---|---|
node() const | ::tensorflow::Node * |
operator::tensorflow::Input() const | |
operator::tensorflow::Output() const |
Public static functions | |
---|---|
Container(StringPiece x) | |
SharedName(StringPiece x) |
Structs | |
---|---|
tensorflow::ops::ConditionalAccumulator::Attrs | Optional attribute setters for ConditionalAccumulator. |
::tensorflow::Output handle
ConditionalAccumulator( const ::tensorflow::Scope & scope, DataType dtype, PartialTensorShape shape )
ConditionalAccumulator( const ::tensorflow::Scope & scope, DataType dtype, PartialTensorShape shape, const ConditionalAccumulator::Attrs & attrs )
::tensorflow::Node * node() const
operator::tensorflow::Input() const
operator::tensorflow::Output() const
Attrs Container( StringPiece x )
Attrs SharedName( StringPiece x )
© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/conditional-accumulator.html