diff options
Diffstat (limited to 'delegate/test/ReduceTestHelper.hpp')
-rw-r--r-- | delegate/test/ReduceTestHelper.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
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) { |