diff --git a/inference-engine/src/legacy_api/include/legacy/ngraph_ops/gather_tree_ie.hpp b/inference-engine/src/legacy_api/include/legacy/ngraph_ops/gather_tree_ie.hpp index 02d914c9bda..89350cb8a17 100644 --- a/inference-engine/src/legacy_api/include/legacy/ngraph_ops/gather_tree_ie.hpp +++ b/inference-engine/src/legacy_api/include/legacy/ngraph_ops/gather_tree_ie.hpp @@ -32,6 +32,7 @@ public: const Output& end_token); void validate_and_infer_types() override; + bool visit_attributes(AttributeVisitor& visitor) override; std::shared_ptr clone_with_new_inputs(const OutputVector& new_args) const override; }; diff --git a/inference-engine/src/legacy_api/include/legacy/ngraph_ops/hard_sigmoid_ie.hpp b/inference-engine/src/legacy_api/include/legacy/ngraph_ops/hard_sigmoid_ie.hpp index ac8489bf420..0361aae89c7 100644 --- a/inference-engine/src/legacy_api/include/legacy/ngraph_ops/hard_sigmoid_ie.hpp +++ b/inference-engine/src/legacy_api/include/legacy/ngraph_ops/hard_sigmoid_ie.hpp @@ -27,6 +27,7 @@ public: std::shared_ptr clone_with_new_inputs(const OutputVector& new_args) const override; void validate_and_infer_types() override; + bool visit_attributes(AttributeVisitor& visitor) override; float get_alpha() const { return m_alpha; } void set_alpha(float alpha) { m_alpha = alpha; } diff --git a/inference-engine/src/legacy_api/include/legacy/ngraph_ops/interp.hpp b/inference-engine/src/legacy_api/include/legacy/ngraph_ops/interp.hpp index d7a7dbd1daf..df19fdc0f95 100644 --- a/inference-engine/src/legacy_api/include/legacy/ngraph_ops/interp.hpp +++ b/inference-engine/src/legacy_api/include/legacy/ngraph_ops/interp.hpp @@ -37,6 +37,8 @@ public: void validate_and_infer_types() override; + bool visit_attributes(AttributeVisitor& visitor) override; + std::shared_ptr clone_with_new_inputs(const OutputVector& new_args) const override; InterpolateIEAttrs get_attrs() { return m_attrs; } diff --git a/inference-engine/src/legacy_api/src/ngraph_ops/gather_tree_ie.cpp b/inference-engine/src/legacy_api/src/ngraph_ops/gather_tree_ie.cpp index 0e3c04c13de..dcb4dab38e1 100644 --- a/inference-engine/src/legacy_api/src/ngraph_ops/gather_tree_ie.cpp +++ b/inference-engine/src/legacy_api/src/ngraph_ops/gather_tree_ie.cpp @@ -63,3 +63,7 @@ void op::GatherTreeIE::validate_and_infer_types() { const auto& step_ids_et = get_input_element_type(0); set_output_type(0, step_ids_et, step_ids_rank); } + +bool ngraph::op::GatherTreeIE::visit_attributes(AttributeVisitor& visitor) { + return true; +} diff --git a/inference-engine/src/legacy_api/src/ngraph_ops/hard_sigmoid_ie.cpp b/inference-engine/src/legacy_api/src/ngraph_ops/hard_sigmoid_ie.cpp index d85796c5903..4ccc0e8e60a 100644 --- a/inference-engine/src/legacy_api/src/ngraph_ops/hard_sigmoid_ie.cpp +++ b/inference-engine/src/legacy_api/src/ngraph_ops/hard_sigmoid_ie.cpp @@ -35,3 +35,7 @@ shared_ptr op::HardSigmoid_IE::clone_with_new_inputs(const OutputVector& n check_new_args_count(this, new_args); return make_shared(new_args.at(0), m_alpha, m_beta); } + +bool op::HardSigmoid_IE::visit_attributes(AttributeVisitor& visitor) { + return true; +} diff --git a/inference-engine/src/legacy_api/src/ngraph_ops/interp.cpp b/inference-engine/src/legacy_api/src/ngraph_ops/interp.cpp index 19ffed8697c..f07d73fc04c 100644 --- a/inference-engine/src/legacy_api/src/ngraph_ops/interp.cpp +++ b/inference-engine/src/legacy_api/src/ngraph_ops/interp.cpp @@ -67,6 +67,16 @@ shared_ptr op::Interp::clone_with_new_inputs(const OutputVector& new_args) return make_shared(new_args.at(0), m_attrs); } +bool op::Interp::visit_attributes(AttributeVisitor& visitor) +{ + visitor.on_attribute("align_corners", m_attrs.align_corners); + visitor.on_attribute("width", m_attrs.width); + visitor.on_attribute("height", m_attrs.height); + visitor.on_attribute("pad_beg", m_attrs.pad_beg); + visitor.on_attribute("pad_end", m_attrs.pad_end); + return true; +} + constexpr NodeTypeInfo op::ResampleV2::type_info; op::ResampleV2::ResampleV2(const Output& image, const Output& output_shape,