IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed visitor for Interpolate-1 and Interpolate-4 (#2051)

Browse Source 
* Fixed visitor for Interpolate-1 and Interpolate-4

* Code style fix

* Remove unnecessary changes

* Fixed compilation on Linux for Atttribute visitor of vector<size_t>

* Added unit test for IE IR Reader for Interpolate-4

* Updated unit test for IR Reader for Interpolate-4

* Updated unit test
This commit is contained in:
Evgeny Lazarev
2020-09-03 16:00:46 +03:00
committed by GitHub
parent d9e9abcf50
commit f60b46f3d4
5 changed files with 198 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
ngraph/core/include/ngraph/axis_set.hpp
View File

@@ -60,6 +60,7 @@ namespace ngraph
void set(const std::vector<int64_t>& value) override;
static constexpr DiscreteTypeInfo type_info{"AttributeAdapter<AxisSet>", 0};
const DiscreteTypeInfo& get_type_info() const override { return type_info; }
operator AxisSet&() { return m_ref; }
protected:
AxisSet& m_ref;
std::vector<int64_t> m_buffer;

4
ngraph/core/src/op/interpolate.cpp
View File

@@ -38,10 +38,10 @@ op::v0::Interpolate::Interpolate(const Output<Node>& image,
bool op::v0::Interpolate::visit_attributes(AttributeVisitor& visitor)
{
visitor.on_attribute("axes", m_attrs.axes);
visitor.on_attribute("mode", m_attrs.mode);
visitor.on_attribute("align_corners", m_attrs.align_corners);
visitor.on_attribute("antialias", m_attrs.antialias);
visitor.on_attribute("axes", m_attrs.axes);
visitor.on_attribute("mode", m_attrs.mode);
visitor.on_attribute("pads_begin", m_attrs.pads_begin);
visitor.on_attribute("pads_end", m_attrs.pads_end);
return true;