diff options
Diffstat (limited to 'verif/frameworks')
-rwxr-xr-x | verif/frameworks/tosa_verif_framework_generator.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/verif/frameworks/tosa_verif_framework_generator.py b/verif/frameworks/tosa_verif_framework_generator.py index 742f991..c93fc88 100755 --- a/verif/frameworks/tosa_verif_framework_generator.py +++ b/verif/frameworks/tosa_verif_framework_generator.py @@ -1204,7 +1204,7 @@ def run_unit_test( if tflite_model_filename: tflite_result_npy_filename = "tflite_result.npy" - ops_with_optimized_only_kernel = ["elu", "ceil", "gather"] + ops_with_optimized_only_kernel = ["elu", "ceil", "gather", "rfft2d"] if args.tflite_kernel_mode == "optimized" or ( op_name in ops_with_optimized_only_kernel |