From 97381e0b63129befd2ef9ef219db39e2c294413b Mon Sep 17 00:00:00 2001
From: Pawel Raasz <pawel.raasz@intel.com>
Date: Wed, 22 Nov 2023 09:41:53 +0100
Subject: [PATCH] Correct assertion in PlainTensor member functions (#21199)

- PlainTensor::size
- PlainTensor::stride
---
 src/plugins/intel_cpu/src/utils/plain_tensor.hpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/plugins/intel_cpu/src/utils/plain_tensor.hpp b/src/plugins/intel_cpu/src/utils/plain_tensor.hpp
index 01ea872f27dd06..d84da1031571d0 100644
--- a/src/plugins/intel_cpu/src/utils/plain_tensor.hpp
+++ b/src/plugins/intel_cpu/src/utils/plain_tensor.hpp
@@ -131,11 +131,11 @@ struct PlainTensor {
     size_t size(int i) const {
         if (i < 0)
             i += m_rank;
-        assert(i < m_rank);
+        assert(static_cast<typename std::make_unsigned<decltype(i)>::type>(i) < m_rank);
         return m_dims[i];
     }
     size_t stride(int i) const {
-        assert(i < m_rank);
+        assert(i >= 0 && static_cast<typename std::make_unsigned<decltype(i)>::type>(i) < m_rank);
         return m_strides[i];
     }
     PlainTensor(MemoryPtr mem) {