diff options
Diffstat (limited to 'verif')
| -rwxr-xr-x | verif/frameworks/tosa_verif_framework_generator.py | 2 | ||||
| -rw-r--r-- | verif/generator/tosa_test_gen.py | 8 |
2 files changed, 5 insertions, 5 deletions
diff --git a/verif/frameworks/tosa_verif_framework_generator.py b/verif/frameworks/tosa_verif_framework_generator.py index 222376e..8457d92 100755 --- a/verif/frameworks/tosa_verif_framework_generator.py +++ b/verif/frameworks/tosa_verif_framework_generator.py @@ -23,7 +23,7 @@ from frameworks.write_test_json import write_test_json # noqa: E402 from frameworks.arg_gen import ArgGen # noqa: E402 from frameworks.tensor_gen import TGen # noqa: E402 from frameworks.test_builder import TBuilder # noqa: E402 -from frameworks.test_gen_utils import ( +from frameworks.test_gen_utils import ( # noqa: E402 QuantType, get_tf_dtype, get_shape_str, diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index b1f9938..64f0c5e 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -3702,7 +3702,7 @@ class TosaTestGen: raise Exception("Unknown dtype, cannot convert to string: {}".format(t)) def typeWidth(self, t): - """ Get the datatype width for integer types""" + """Get the datatype width for integer types""" if t == DType.INT4: return 4 elif t == DType.INT8: @@ -5921,7 +5921,7 @@ class TosaTestGen: if (divisor_arr == 0).any(): continue - if (dividend_arr == -(2 ** 31)).any() and (divisor_arr == -1).any(): + if (dividend_arr == -(2**31)).any() and (divisor_arr == -1).any(): continue break @@ -5980,8 +5980,8 @@ class TosaTestGen: else: result_arr = a_arr_64 * b_arr_64 - if (result_arr > -(2 ** 31)).all() and ( - result_arr <= ((2 ** 31) - 1) + if (result_arr > -(2**31)).all() and ( + result_arr <= ((2**31) - 1) ).all(): break |