19 #include "ngraph/op/op.hpp"
A set of axes.
Definition: axis_set.hpp:31
A handle for one of a node's outputs.
Definition: node_output.hpp:42
Root of all actual ops.
Definition: op.hpp:29
Abstract base class for arithmetic reduction operations, i.e., operations where chosen axes of the in...
Definition: arithmetic_reduction.hpp:31
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
bool reduction_axes_constant() const
void set_reduction_axes(const AxisSet &reduction_axes)
Change the reduction axes.
const AxisSet get_reduction_axes() const
ArithmeticReduction()
Constructs an arithmetic reduction operation.
ArithmeticReduction(const Output< Node > &arg, const Output< Node > &reduction_axes)
Constructs an arithmetic reduction operation.
ArithmeticReduction(const Output< Node > &arg, const AxisSet &reduction_axes)
Constructs an arithmetic reduction operation.
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28