From d99d18304a21282d09ebf5d327047024169b8c49 Mon Sep 17 00:00:00 2001 From: Liubov Batanina Date: Fri, 6 Dec 2019 09:56:21 +0300 Subject: [PATCH] Slice v1 op --- modules/dnn/src/layers/prior_box_layer.cpp | 6 ++++-- modules/dnn/src/layers/slice_layer.cpp | 5 +++-- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/modules/dnn/src/layers/prior_box_layer.cpp b/modules/dnn/src/layers/prior_box_layer.cpp index fd37004da3..c38e61329b 100644 --- a/modules/dnn/src/layers/prior_box_layer.cpp +++ b/modules/dnn/src/layers/prior_box_layer.cpp @@ -569,8 +569,10 @@ public: auto upper_bounds = std::make_shared(ngraph::element::i64, ngraph::Shape{1}, std::vector{4}); auto strides = std::make_shared(ngraph::element::i64, ngraph::Shape{1}, std::vector{1}); - auto slice_layer = std::make_shared(layer_shape, lower_bounds, upper_bounds, strides); - auto slice_image = std::make_shared(image_shape, lower_bounds, upper_bounds, strides); + auto slice_layer = std::make_shared(layer_shape, + lower_bounds, upper_bounds, strides, std::vector{}, std::vector{}); + auto slice_image = std::make_shared(image_shape, + lower_bounds, upper_bounds, strides, std::vector{}, std::vector{}); if (_explicitSizes) { diff --git a/modules/dnn/src/layers/slice_layer.cpp b/modules/dnn/src/layers/slice_layer.cpp index ea2562380f..d7cafd7f93 100644 --- a/modules/dnn/src/layers/slice_layer.cpp +++ b/modules/dnn/src/layers/slice_layer.cpp @@ -341,8 +341,9 @@ public: auto strides = std::make_shared(ngraph::element::i64, ngraph::Shape{dims.size()}, std::vector((int64_t)dims.size(), 1)); - auto slice = std::make_shared(ieInpNode, lower_bounds, upper_bounds, - strides, ngraph::AxisSet{}, ngraph::AxisSet{}); + auto slice = std::make_shared(ieInpNode, + lower_bounds, upper_bounds, strides, std::vector{}, std::vector{}); + return Ptr(new InfEngineNgraphNode(slice)); } #endif // HAVE_DNN_NGRAPH