update per review

openvinotoolkit · May 4, 2021 · 50f4cb5 · 50f4cb5
1 parent 4425096
commit 50f4cb5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/inference-engine/src/mkldnn_plugin/nodes/mkldnn_mvn_node.cpp b/inference-engine/src/mkldnn_plugin/nodes/mkldnn_mvn_node.cpp
@@ -624,7 +624,8 @@ void MKLDNNMVNNode::getSupportedDescriptors() {
     if (getChildEdges().empty())
         IE_THROW() << errPrefix << "has incorrect number of output edges.";
 
-    const auto& numOfDims = getParentEdgeAt(0)->getDims().ndims();
+    SizeVector inputShape = getParentEdgeAt(0)->getDims().ToSizeVector();
+    const auto& numOfDims = inputShape.size();
     if (numOfDims < 1 || numOfDims > 5)
         IE_THROW() << errPrefix << "doesn't support input with size of dimensions: " << numOfDims;
 
@@ -644,7 +645,6 @@ void MKLDNNMVNNode::getSupportedDescriptors() {
         epsMode_ = outsideSqrt;
     }
 
-    SizeVector inputShape = getParentEdgeAt(0)->getDims().ToSizeVector();
     transformTo5DCase(inputShape);
 }