1 files changed, 36 insertions, 0 deletions
diff --git a/verif/frameworks/tosa_verif_framework_generator.py b/verif/frameworks/tosa_verif_framework_generator.py
index 760def6..26af5dd 100755
--- a/verif/frameworks/tosa_verif_framework_generator.py
+++ b/verif/frameworks/tosa_verif_framework_generator.py
@@ -807,6 +807,42 @@ TF_OP_LIST = {
             ]
         },
     },
+    "while": {
+        "operands": (1, 0),
+        "build_fcn": (TBuilder.While, TGen.tgBasic, ArgGen.agNone),
+        "types": {
+            "tflite": list(TYPE_F),
+        },
+    },
+    "lstm": {
+        "operands": (1, 0),
+        "build_fcn": (TBuilder.LSTM, TGen.tgRecurrent, ArgGen.agNone),
+        "types": {
+            "tflite": [
+                tf.float32,
+                # tf.int32
+            ]
+        },
+    },
+    "gru": {
+        "operands": (1, 0),
+        "build_fcn": (TBuilder.GRU, TGen.tgRecurrent, ArgGen.agNone),
+        "types": {
+            "tflite": [
+                tf.float32,
+                # tf.int32
+            ]
+        },
+    },
+    "rnn": {
+        "operands": (1, 0),
+        "build_fcn": (TBuilder.RNN, TGen.tgRecurrent, ArgGen.agNone),
+        "types": {
+            "tflite": [
+                tf.float32,
+            ]
+        },
+    },
 }
 
 # Shapes to be tested; default can be overwritten