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

[IE][VPU]: dynamic hswish and gather downgrade (#9915)

Browse Source 
* Dynamic HSwish
* Gather8 to Gather7 downgrade
This commit is contained in:
Maksim Doronin 2022-01-26 16:28:03 +03:00 committed by GitHub
parent 2ea52bae6e
commit e46b629950
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/plugins/intel_vpu/common/src/ngraph/transformations/dynamic_to_static_shape.cpp
View File

@ -35,6 +35,7 @@
#include <ngraph/validation_util.hpp>
#include "ngraph/opsets/opset5.hpp"
#include "ngraph/opsets/opset6.hpp"
#include "ngraph/opsets/opset8.hpp"
namespace vpu {
@ -129,6 +130,7 @@ const Transformations& getDefaultTransformations() {
{ngraph::opset3::LogicalNot::get_type_info_static(), dynamicToStaticUnaryElementwise},
{ngraph::opset3::Abs::get_type_info_static(), dynamicToStaticUnaryElementwise},
{ngraph::opset5::ScatterElementsUpdate::get_type_info_static(), dynamicToStaticUnaryElementwise},
{ngraph::opset8::HSwish::get_type_info_static(), dynamicToStaticUnaryElementwise},
{ngraph::opset3::StridedSlice::get_type_info_static(), dynamicToStaticShapeStridedSlice},
{ngraph::opset3::Squeeze::get_type_info_static(), dynamicToStaticShapeSqueeze},
{ngraph::opset3::Gather::get_type_info_static(), dynamicToStaticShapeGather},

1
src/plugins/intel_vpu/graph_transformer/src/frontend/frontend.cpp
View File

@ -181,6 +181,7 @@ ie::CNNNetwork FrontEnd::convertNetwork(ie::CNNNetwork& network) {
manager.register_pass<ngraph::pass::ConvertNMS1ToNMS5>();
manager.register_pass<ngraph::pass::ConvertNMS3ToNMS5>();
manager.register_pass<ngraph::pass::ConvertNMS4ToNMS5>();
manager.register_pass<ngraph::pass::ConvertGather8ToGather7>();
manager.register_pass<ngraph::pass::ConvertGather7ToGather1>();
manager.register_pass<vpu::ConvertGatherND8ToGatherND5>();
manager.register_pass<vpu::MergeGatherGatherElements>();