21 #include "ngraph/deprecated.hpp"
22 #include "ngraph/node.hpp"
23 #include "ngraph/op/op.hpp"
34 static constexpr NodeTypeInfo type_info{
"BatchNormInference", 0};
35 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
44 const Output<Node>& gamma,
45 const Output<Node>& beta,
46 const Output<Node>& mean,
47 const Output<Node>& variance,
54 double get_eps_value()
const {
return m_epsilon; }
55 void set_eps_value(
double epsilon) { m_epsilon = epsilon; }
57 clone_with_new_inputs(
const OutputVector& new_args)
const override;
60 static constexpr
size_t INPUT_GAMMA = 0;
61 static constexpr
size_t INPUT_BETA = 1;
62 static constexpr
size_t INPUT_DATA = 2;
63 static constexpr
size_t INPUT_MEAN = 3;
64 static constexpr
size_t INPUT_VARIANCE = 4;
69 using v0::BatchNormInference;
Definition: batch_norm.hpp:32
const NodeTypeInfo & get_type_info() const override
Definition: batch_norm.hpp:35
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
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:70
BatchNormInference(const Output< Node > &input, const Output< Node > &gamma, const Output< Node > &beta, const Output< Node > &mean, const Output< Node > &variance, double epsilon)
Root of all actual ops.
Definition: op.hpp:29