[NPU][Dynamism] - Fix L0 dynamic shape inference - getBatchSizeForNode

DariaMityagina · Jul 30, 2024 · 650e8e1 · 650e8e1
1 parent 127ddce
commit 650e8e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/plugins/intel_npu/src/backend/src/zero_infer_request.cpp b/src/plugins/intel_npu/src/backend/src/zero_infer_request.cpp
@@ -69,7 +69,7 @@ std::optional<size_t> getBatchSizeForNode(const IONodeDescriptor& nodeDescriptor
         return std::nullopt;
     }
 
-    const std::vector<size_t>& ovDimensions = nodeDescriptor.originalShape.get_shape();
+    const std::vector<size_t>& ovDimensions = nodeDescriptor.originalShape.get_max_shape();
 
     if (ovDimensions[BATCH_AXIS] == zeDescriptor.info.dims[BATCH_AXIS] &&
         ovDimensions[BATCH_AXIS] != DEFAULT_BATCH_SIZE) {