diff options
Diffstat (limited to 'verif/frameworks/tosa_verif_framework_generator.py')
-rwxr-xr-x | verif/frameworks/tosa_verif_framework_generator.py | 36 |
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 |