diff --git a/src/plugins/intel_cpu/src/nodes/matrix_nms.cpp b/src/plugins/intel_cpu/src/nodes/matrix_nms.cpp index fe825800f34..2cc63d46c78 100644 --- a/src/plugins/intel_cpu/src/nodes/matrix_nms.cpp +++ b/src/plugins/intel_cpu/src/nodes/matrix_nms.cpp @@ -376,7 +376,7 @@ void MatrixNms::execute(dnnl::stream strm) { // NMS-alike nodes are always transformed to NMSIEInternal node in case of legacy api, for compatibility. // And on the other hand in case of api 2.0, keep them internal dynamic for better performance and functionality. - if (isDynamicNode()) { + if (!m_outStaticShape) { size_t totalBox = std::accumulate(m_numPerBatch.begin(), m_numPerBatch.end(), size_t(0)); redefineOutputMemory({{totalBox, 6}, {totalBox, 1}, {m_numBatches}}); } diff --git a/src/plugins/intel_cpu/src/nodes/multiclass_nms.cpp b/src/plugins/intel_cpu/src/nodes/multiclass_nms.cpp index 805740ca4ed..e7100b5d708 100644 --- a/src/plugins/intel_cpu/src/nodes/multiclass_nms.cpp +++ b/src/plugins/intel_cpu/src/nodes/multiclass_nms.cpp @@ -166,7 +166,7 @@ void MultiClassNms::prepareParams() { } else { m_numBatches = boxes_dims[0]; } - m_numBoxes = shared ? boxes_dims[1] : boxes_dims[1]; + m_numBoxes = boxes_dims[1]; m_numClasses = shared ? scores_dims[1] : scores_dims[0]; int max_output_boxes_per_class = 0;