Switch to ov shape_utils (#818)

openvinotoolkit · Jan 10, 2024 · 3d321c9 · 3d321c9
1 parent b8ff139
commit 3d321c9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/modules/nvidia_plugin/src/ops/mvn.cpp b/modules/nvidia_plugin/src/ops/mvn.cpp
@@ -4,10 +4,10 @@
 
 #include "mvn.hpp"
 
+#include "openvino/core/shape_util.hpp"
 #include "openvino/core/type.hpp"
 #include "openvino/op/constant.hpp"
 #include "openvino/op/mvn.hpp"
-#include "ngraph/shape_util.hpp"
 
 #include <cuda/descriptor_utils.hpp>
 #include <cuda_operation_registry.hpp>
@@ -224,7 +224,7 @@ ov::Shape MvnOp::makeReducedShape(const ov::Node& node) {
             }
             axes.emplace(static_cast<size_t>((v + size) % size));
         }
-        reducedShape = ngraph::reduce(reducedShape, axes, true);
+        reducedShape = ov::util::reduce_keep_dims(reducedShape, axes);
         if (reducedShape == node.get_input_shape(0)) return {};
         return reducedShape;
     }