diff --git a/modules/dnn/src/layers/lrn_layer.cpp b/modules/dnn/src/layers/lrn_layer.cpp index 6c2ab34a0b..5926d9053f 100644 --- a/modules/dnn/src/layers/lrn_layer.cpp +++ b/modules/dnn/src/layers/lrn_layer.cpp @@ -96,7 +96,7 @@ public: return bias == (int)bias; } if (backendId == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH) { - return type == CHANNEL_NRM && bias == (int)bias; + return bias == (int)bias; } return backendId == DNN_BACKEND_OPENCV || backendId == DNN_BACKEND_HALIDE; } @@ -412,7 +412,15 @@ public: alphaSize *= (type == SPATIAL_NRM ? size*size : size); auto& ieInpNode = nodes[0].dynamicCast()->node; - auto lrn = std::make_shared(ieInpNode, (double)alphaSize, (double)beta, (double)bias, (size_t)size); + std::vector axes; + if (type != SPATIAL_NRM) { + axes = {1}; + } else { + axes.resize(ieInpNode->get_shape().size() - 2); + std::iota(axes.begin(), axes.end(), 2); + } + auto ngraph_axes = std::make_shared(ngraph::element::i64, ngraph::Shape{axes.size()}, axes.data()); + auto lrn = std::make_shared(ieInpNode, ngraph_axes, alphaSize, beta, bias, size); return Ptr(new InfEngineNgraphNode(lrn)); } #endif // HAVE_DNN_NGRAPH