diff options
Diffstat (limited to 'verif/frameworks/tosa_verif_framework_generator.py')
-rwxr-xr-x | verif/frameworks/tosa_verif_framework_generator.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/verif/frameworks/tosa_verif_framework_generator.py b/verif/frameworks/tosa_verif_framework_generator.py index 58df588..ccbe742 100755 --- a/verif/frameworks/tosa_verif_framework_generator.py +++ b/verif/frameworks/tosa_verif_framework_generator.py @@ -860,6 +860,13 @@ TF_OP_LIST = { "tflite": [tf.complex64], }, }, + "broadcastto": { + "operands": (1, 1), + "build_fcn": (TBuilder.BroadcastTo, TGen.tgBroadcastTo, ArgGen.agNone), + "types": { + "tf": TYPE_FIB, + }, + }, } # Shapes to be tested; default can be overwritten |