[IE VPUX][cherry pick #3001] Part of changes: Visitor API deserialization for ops from opset1 (#3558)
This commit is contained in:
committed by
GitHub
parent
39447e08e7
commit
fb1536f2be
@@ -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;
|
||||
};
|
||||
|
||||
@@ -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; }
|
||||
|
||||
@@ -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; }
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user