From 08ee1aaef1963ba75c8978324dfefa8752d95108 Mon Sep 17 00:00:00 2001
From: Idriss Chaouch <idriss.chaouch@arm.com>
Date: Wed, 13 Sep 2023 13:53:37 +0100
Subject: IVGCVSW-8045 ReduceSum Delegate test fails when migrating to
 TensorFlow 2.14

Signed-off-by: Idriss Chaouch <idriss.chaouch@arm.com>
Change-Id: I3a56b2be89ddc64ab61eb04823d4523790affabe
---
 delegate/test/ReduceTestHelper.hpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'delegate')

diff --git a/delegate/test/ReduceTestHelper.hpp b/delegate/test/ReduceTestHelper.hpp
index a268981865..adbbfc3a81 100644
--- a/delegate/test/ReduceTestHelper.hpp
+++ b/delegate/test/ReduceTestHelper.hpp
@@ -49,7 +49,7 @@ std::vector<char> CreateReduceTfLiteModel(tflite::BuiltinOperator reduceOperator
 
     flatbuffers::Offset<tflite::QuantizationParameters> quantizationParameters;
 
-    if (kTfLiteNoQuantizationForQuantized)
+    if (kTfLiteNoQuantizationForQuantized && reduceOperatorCode == BuiltinOperator_REDUCE_PROD)
     {
         if ((quantScale == 1 || quantScale == 0) && quantOffset == 0)
         {
-- 
cgit v1.2.1