From 5bbbc021aae0de6ed27d203fe52ab486b2143cab Mon Sep 17 00:00:00 2001 From: Idriss Chaouch Date: Wed, 4 Oct 2023 12:25:32 +0100 Subject: IVGCVSW-8081 DTS: Fix Unspecified dimension while using ShapeInferenceMethod::ValidateOnly Signed-off-by: Idriss Chaouch Change-Id: Iac4a81ecda50bb179a718522d87caea1fd3760fd --- delegate/classic/src/ClassicDelegateUtils.hpp | 2 +- delegate/opaque/src/OpaqueDelegateUtils.hpp | 2 +- 2 files changed, 2 insertions(+), 2 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 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; } diff --git a/delegate/opaque/src/OpaqueDelegateUtils.hpp b/delegate/opaque/src/OpaqueDelegateUtils.hpp index a636161300..792dc5d28f 100644 --- a/delegate/opaque/src/OpaqueDelegateUtils.hpp +++ b/delegate/opaque/src/OpaqueDelegateUtils.hpp @@ -509,7 +509,7 @@ armnn::TensorInfo GetTensorInfoForTfLiteOpaqueTensor(const TfLiteOpaqueTensor* t { int32_t dim = TfLiteOpaqueTensorDim(tfLiteTensor, i); - if (dim <= 0) + if (dim < 0) { dimensionsSpecificity[i] = false; } -- cgit v1.2.1