19 #include "ngraph/op/op.hpp"
20 #include "ngraph/op/util/logical_reduction.hpp"
38 const 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: logical_reduction_keep_dims.hpp:29
LogicalReductionKeepDims(const Output< Node > &arg, const Output< Node > &reduction_axes, const bool keep_dims=false)
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
bool get_keep_dims() const
Definition: logical_reduction_keep_dims.hpp:47
Abstract base class for logical reduction operations, i.e., operations where chosen axes of the input...
Definition: logical_reduction.hpp:31
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28