21 #include "ngraph/deprecated.hpp"
22 #include "ngraph/node.hpp"
23 #include "ngraph/op/op.hpp"
34 NGRAPH_RTTI_DECLARATION;
53 double get_eps_value()
const {
return m_epsilon; }
54 void set_eps_value(
double epsilon) { m_epsilon = epsilon; }
56 clone_with_new_inputs(
const OutputVector& new_args)
const override;
59 static constexpr
size_t INPUT_GAMMA = 0;
60 static constexpr
size_t INPUT_BETA = 1;
61 static constexpr
size_t INPUT_DATA = 2;
62 static constexpr
size_t INPUT_MEAN = 3;
63 static constexpr
size_t INPUT_VARIANCE = 4;
73 NGRAPH_RTTI_DECLARATION;
92 double get_eps_value()
const {
return m_epsilon; }
93 void set_eps_value(
double epsilon) { m_epsilon = epsilon; }
95 clone_with_new_inputs(
const OutputVector& new_args)
const override;
98 static constexpr
size_t INPUT_DATA = 0;
99 static constexpr
size_t INPUT_GAMMA = 1;
100 static constexpr
size_t INPUT_BETA = 2;
101 static constexpr
size_t INPUT_MEAN = 3;
102 static constexpr
size_t INPUT_VARIANCE = 4;
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
Root of all actual ops.
Definition: op.hpp:29
Definition: batch_norm.hpp:32
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
BatchNormInference(const Output< Node > &input, const Output< Node > &gamma, const Output< Node > &beta, const Output< Node > &mean, const Output< Node > &variance, double epsilon)
Definition: batch_norm.hpp:71
BatchNormInference(const Output< Node > &input, const Output< Node > &gamma, const Output< Node > &beta, const Output< Node > &mean, const Output< Node > &variance, double epsilon)
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28