19 #include "ngraph/op/op.hpp"
20 #include "ngraph/op/util/arithmetic_reduction.hpp"
38 bool keep_dims =
false);
48 void set_keep_dims(
bool keep_dims) { m_keep_dims = keep_dims; }
50 bool m_keep_dims =
false;
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:71
A handle for one of a node's outputs.
Definition: node_output.hpp:42
Definition: arithmetic_reductions_keep_dims.hpp:29
bool get_keep_dims() const
Definition: arithmetic_reductions_keep_dims.hpp:47
ArithmeticReductionKeepDims(const Output< Node > &arg, const Output< Node > &reduction_axes, bool keep_dims=false)
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
Abstract base class for arithmetic reduction operations, i.e., operations where chosen axes of the in...
Definition: arithmetic_reduction.hpp:31
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28