commit 41a04feacf0fecc14b1d0c50c39a1dc3e3710c4d
author TatWai Chong <tatwai.chong@arm.com> Thu Nov 03 21:44:32 2022 +0000
committer Eric Kunze <eric.kunze@arm.com> Fri Nov 11 23:56:46 2022 +0000
tree 328263a180a4be0564f498cade694b8b6f3568c9
parent 24dbc420aae556649f50e645bd94489dab2cc75a

Add framework test for tfl.prelu

Signed-off-by: TatWai Chong <tatwai.chong@arm.com>
Change-Id: Ib894e50f4b869a3f4936885704995a417ea25eef

verif/frameworks/tosa_verif_framework_generator.py

diff --git a/verif/frameworks/tosa_verif_framework_generator.py b/verif/frameworks/tosa_verif_framework_generator.py
index 4c710bd..5cc764f 100755
--- a/verif/frameworks/tosa_verif_framework_generator.py
+++ b/verif/frameworks/tosa_verif_framework_generator.py
@@ -140,6 +140,13 @@
             ),
         },
     },
+    "prelu": {
+        "operands": (1, 0),
+        "build_fcn": (TBuilder.Prelu, TGen.tgBasic, ArgGen.agNone),
+        "types": {
+            "tflite": list(TYPE_F + [QuantType.ALL_U8, QuantType.ALL_I8]),
+        },
+    },
     "gelu": {
         "operands": (1, 0),
         "build_fcn": (TBuilder.Gelu, TGen.tgBasic, ArgGen.agNone),