[IE VPUX][cherry pick #3001] Part of changes: Visitor API deserialization for ops from opset1 (#3558)

This commit is contained in:
Nikita Kudriavtsev
2020-12-11 18:06:58 +03:00
committed by GitHub
parent 39447e08e7
commit fb1536f2be
6 changed files with 22 additions and 0 deletions

View File

@@ -32,6 +32,7 @@ public:
const Output<Node>& end_token);
void validate_and_infer_types() override;
bool visit_attributes(AttributeVisitor& visitor) override;
std::shared_ptr<Node> clone_with_new_inputs(const OutputVector& new_args) const override;
};

View File

@@ -27,6 +27,7 @@ public:
std::shared_ptr<Node> 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; }

View File

@@ -37,6 +37,8 @@ public:
void validate_and_infer_types() override;
bool visit_attributes(AttributeVisitor& visitor) override;
std::shared_ptr<Node> clone_with_new_inputs(const OutputVector& new_args) const override;
InterpolateIEAttrs get_attrs() { return m_attrs; }

View File

@@ -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;
}

View File

@@ -35,3 +35,7 @@ shared_ptr<Node> op::HardSigmoid_IE::clone_with_new_inputs(const OutputVector& n
check_new_args_count(this, new_args);
return make_shared<op::HardSigmoid_IE>(new_args.at(0), m_alpha, m_beta);
}
bool op::HardSigmoid_IE::visit_attributes(AttributeVisitor& visitor) {
return true;
}

View File

@@ -67,6 +67,16 @@ shared_ptr<Node> op::Interp::clone_with_new_inputs(const OutputVector& new_args)
return make_shared<Interp>(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<Node>& image, const Output<Node>& output_shape,