From bb5b628cce18c3b947c12aef6df5063244b8d1ea Mon Sep 17 00:00:00 2001 From: Roman Kazantsev Date: Fri, 6 Nov 2020 09:51:40 +0300 Subject: [PATCH] Use explicit opset of Unsqueeze from nGraph backporting commit eb24575e2ce6ae56613fe4b9709ea55b4d8a228e --- modules/dnn/src/layers/prior_box_layer.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/dnn/src/layers/prior_box_layer.cpp b/modules/dnn/src/layers/prior_box_layer.cpp index 7385afd3b0..dd39ce4417 100644 --- a/modules/dnn/src/layers/prior_box_layer.cpp +++ b/modules/dnn/src/layers/prior_box_layer.cpp @@ -595,7 +595,7 @@ public: auto priorBox = std::make_shared(slice_layer, slice_image, attrs); auto axis = std::make_shared(ngraph::element::i64, ngraph::Shape{1}, std::vector{0}); - auto unsqueeze = std::make_shared(priorBox, axis); + auto unsqueeze = std::make_shared(priorBox, axis); return Ptr(new InfEngineNgraphNode(unsqueeze)); } else @@ -616,7 +616,7 @@ public: auto priorBox = std::make_shared(slice_layer, slice_image, attrs); auto axis = std::make_shared(ngraph::element::i64, ngraph::Shape{1}, std::vector{0}); - auto unsqueeze = std::make_shared(priorBox, axis); + auto unsqueeze = std::make_shared(priorBox, axis); return Ptr(new InfEngineNgraphNode(unsqueeze)); } }