diff options
author | Idriss Chaouch <idriss.chaouch@arm.com> | 2023-10-04 12:25:32 +0100 |
---|---|---|
committer | Idriss Chaouch <idriss.chaouch@arm.com> | 2023-10-06 11:04:57 +0000 |
commit | 5bbbc021aae0de6ed27d203fe52ab486b2143cab (patch) | |
tree | 53a858217d5b49c820f900c91de2a215d7762136 /delegate/classic | |
parent | 19ad816539037a095749cdd41852292a3d87dd2b (diff) | |
download | armnn-5bbbc021aae0de6ed27d203fe52ab486b2143cab.tar.gz |
IVGCVSW-8081 DTS: Fix Unspecified dimension while using ShapeInferenceMethod::ValidateOnly
Signed-off-by: Idriss Chaouch <idriss.chaouch@arm.com>
Change-Id: Iac4a81ecda50bb179a718522d87caea1fd3760fd
Diffstat (limited to 'delegate/classic')
-rw-r--r-- | delegate/classic/src/ClassicDelegateUtils.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/delegate/classic/src/ClassicDelegateUtils.hpp b/delegate/classic/src/ClassicDelegateUtils.hpp index d378c6c3b0..f5ff32a3df 100644 --- a/delegate/classic/src/ClassicDelegateUtils.hpp +++ b/delegate/classic/src/ClassicDelegateUtils.hpp @@ -437,7 +437,7 @@ armnn::TensorInfo GetTensorInfoForTfLiteTensor(const TfLiteTensor& tfLiteTensor, 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; } |