diff options
Diffstat (limited to 'verif/frameworks/tosa_verif_framework_compiler_runner.py')
| -rwxr-xr-x | verif/frameworks/tosa_verif_framework_compiler_runner.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/verif/frameworks/tosa_verif_framework_compiler_runner.py b/verif/frameworks/tosa_verif_framework_compiler_runner.py index ce9b253..56daa51 100755 --- a/verif/frameworks/tosa_verif_framework_compiler_runner.py +++ b/verif/frameworks/tosa_verif_framework_compiler_runner.py @@ -695,7 +695,11 @@ def run_test(args, test_path, framework): tf_result = tf_result.astype(np.int8) elif tf_result.dtype == np.uint8: tf_result = tf_result.astype(np.uint8) - elif tf_result.dtype == np.int16 or tf_result.dtype == np.int64: + elif tf_result.dtype == np.int16: + tf_result = tf_result.astype(np.int16) + elif tf_result.dtype == np.uint16: + tf_result = tf_result.astype(np.uint16) + elif tf_result.dtype == np.int64: tf_result = tf_result.astype(np.int32) # For now, search for the first output from ref_model |