19 #include "ngraph/node.hpp"
20 #include "ngraph/op/op.hpp"
21 #include "ngraph/op/util/fused_op.hpp"
27 NGRAPH_SUPPRESS_DEPRECATED_START
33 class NGRAPH_API
MVN :
public ngraph::op::util::FusedOp
36 NGRAPH_RTTI_DECLARATION;
50 bool across_channels =
true,
51 bool normalize_variance =
true,
65 bool normalize_variance =
true,
68 virtual OutputVector decompose_op()
const override;
70 virtual void validate_and_infer_types()
override;
74 virtual std::shared_ptr<Node>
75 clone_with_new_inputs(
const OutputVector& new_args)
const override;
77 double get_eps()
const {
return m_eps; }
78 bool get_across_channels()
const {
return m_across_channels; }
79 bool get_normalize_variance()
const {
return m_normalize_variance; }
80 AxisSet get_reduction_axes()
const {
return m_reduction_axes; }
81 void set_reduction_axes(AxisSet axes) { m_reduction_axes = axes; }
84 bool m_across_channels;
85 bool m_normalize_variance;
86 AxisSet m_reduction_axes;
91 NGRAPH_SUPPRESS_DEPRECATED_END
103 std::ostream& operator<<(std::ostream& s,
const MVNEpsMode& type);
112 NGRAPH_RTTI_DECLARATION;
127 bool normalize_variance,
134 std::shared_ptr<Node>
135 clone_with_new_inputs(
const OutputVector& new_args)
const override;
137 float get_eps()
const {
return m_eps; }
138 bool get_normalize_variance()
const {
return m_normalize_variance; }
139 MVNEpsMode get_eps_mode()
const {
return m_eps_mode; }
141 bool m_normalize_variance =
true;
142 float m_eps = (float)1e-6;
143 MVNEpsMode m_eps_mode = MVNEpsMode::INSIDE_SQRT;
158 static constexpr
DiscreteTypeInfo type_info{
"AttributeAdapter<op::MVNEpsMode>", 0};
159 const DiscreteTypeInfo& get_type_info()
const override {
return type_info; }
An AttributeAdapter "captures" an attribute as an AT& and makes it available as a ValueAccessor<VAT>.
Definition: attribute_adapter.hpp:171
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:71
A set of axes.
Definition: axis_set.hpp:31
Access an enum via a string.
Definition: attribute_adapter.hpp:178
A handle for one of a node's outputs.
Definition: node_output.hpp:42
Root of all actual ops.
Definition: op.hpp:29
Operator performing Mean Variance Normalization.
Definition: mvn.hpp:34
MVN(const Output< Node > &data, AxisSet reduction_axes, bool normalize_variance=true, double eps=1e-9)
Constructs an MVN operation.
MVN(const Output< Node > &data, bool across_channels=true, bool normalize_variance=true, double eps=1e-9)
Constructs an MVN operation.
Operator performing Mean Variance Normalization.
Definition: mvn.hpp:110
MVN(const Output< Node > &data, const Output< Node > &reduction_axes, bool normalize_variance, float eps, MVNEpsMode eps_mode)
Constructs an MVN operation.
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
MVNEpsMode
Specifies how eps is applied in MVN.
Definition: mvn.hpp:95
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28