commit 5bbbc021aae0de6ed27d203fe52ab486b2143cab
author Idriss Chaouch <idriss.chaouch@arm.com> Wed Oct 04 12:25:32 2023 +0100
committer Idriss Chaouch <idriss.chaouch@arm.com> Fri Oct 06 11:04:57 2023 +0000
tree 53a858217d5b49c820f900c91de2a215d7762136
parent 19ad816539037a095749cdd41852292a3d87dd2b

IVGCVSW-8081 DTS: Fix Unspecified dimension while using ShapeInferenceMethod::ValidateOnly

Signed-off-by: Idriss Chaouch <idriss.chaouch@arm.com>
Change-Id: Iac4a81ecda50bb179a718522d87caea1fd3760fd

delegate/classic/src/ClassicDelegateUtils.hpp

diff --git a/delegate/classic/src/ClassicDelegateUtils.hpp b/delegate/classic/src/ClassicDelegateUtils.hpp
index d378c6c..f5ff32a 100644
--- a/delegate/classic/src/ClassicDelegateUtils.hpp
+++ b/delegate/classic/src/ClassicDelegateUtils.hpp
@@ -437,7 +437,7 @@
         std::vector<unsigned char> dimensionsSpecificity(tensorDimensionSize, true);
         for (int i = 0; i < tensorDimensionSize; ++i) {
             auto dim = tfLiteTensor.dims->data[i];
-            if (dim <= 0)
+            if (dim < 0)
             {
                 dimensionsSpecificity[i] = false;
             }

delegate/opaque/src/OpaqueDelegateUtils.hpp

diff --git a/delegate/opaque/src/OpaqueDelegateUtils.hpp b/delegate/opaque/src/OpaqueDelegateUtils.hpp
index a636161..792dc5d 100644
--- a/delegate/opaque/src/OpaqueDelegateUtils.hpp
+++ b/delegate/opaque/src/OpaqueDelegateUtils.hpp
@@ -509,7 +509,7 @@
         {
             int32_t dim = TfLiteOpaqueTensorDim(tfLiteTensor, i);
 
-            if (dim <= 0)
+            if (dim < 0)
             {
                 dimensionsSpecificity[i] = false;
             }