From 6ad6d6e6c4236317c0835f93b0ff2647e628c29e Mon Sep 17 00:00:00 2001
From: TatWai Chong <tatwai.chong@arm.com>
Date: Fri, 11 Nov 2022 15:28:50 -0800
Subject: Extend reduce_all to test with tflite

Signed-off-by: TatWai Chong <tatwai.chong@arm.com>
Change-Id: Ifb3ed9c448005d6d53c5d0938b5146003a26e8a3
---
 verif/frameworks/tosa_verif_framework_generator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/verif/frameworks/tosa_verif_framework_generator.py b/verif/frameworks/tosa_verif_framework_generator.py
index 931aef6..c11a83d 100755
--- a/verif/frameworks/tosa_verif_framework_generator.py
+++ b/verif/frameworks/tosa_verif_framework_generator.py
@@ -208,7 +208,7 @@ TF_OP_LIST = {
     "reduce_all": {
         "operands": (1, 0),
         "build_fcn": (TBuilder.ReduceAll, TGen.tgBasic, ArgGen.agAxesListKeepdims),
-        "types": {"tf": TYPE_B},
+        "types": TYPE_B,
     },
     "reduce_min": {
         "operands": (1, 0),
-- 
cgit v1.2.1