diff options
| author | Kevin Cheng <kevin.cheng@arm.com> | 2021-07-28 17:19:23 -0700 |
|---|---|---|
| committer | Kevin Cheng <kevin.cheng@arm.com> | 2021-08-12 15:27:03 -0700 |
| commit | a9017401461224b9bc81e7b1c770ca6091e0e3fb (patch) | |
| tree | a9eba569a9ed3862f459b3190787cdbac57da85b /verif/tosa_test_gen.py | |
| parent | 7ffccce9b9a7eeecbad0c0525f1545c2714f8556 (diff) | |
| download | reference_model-a9017401461224b9bc81e7b1c770ca6091e0e3fb.tar.gz | |
Support int4 weights read. Added conv2d int8xint4 in test generation.
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: I61620f160c7dad6aac5fcc3da0a6e97f3bae5b40
Diffstat (limited to 'verif/tosa_test_gen.py')
| -rw-r--r-- | verif/tosa_test_gen.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/verif/tosa_test_gen.py b/verif/tosa_test_gen.py index e375a2a..5c25f8e 100644 --- a/verif/tosa_test_gen.py +++ b/verif/tosa_test_gen.py @@ -923,8 +923,9 @@ class TosaTestGen: if dtype == DType.BOOL: np_dt = np.bool return np.bool_(self.rng.choice(a=[False, True], size=shape)) + # TOSA specific INT4 weight range from -7 to 7 elif dtype == DType.INT4: - return np.int32(self.rng.integers(low=-8, high=8, size=shape)) + return np.int32(self.rng.integers(low=-7, high=8, size=shape)) elif dtype == DType.INT8: return np.int32(self.rng.integers(low=-128, high=128, size=shape)) elif dtype == DType.UINT8: @@ -988,8 +989,9 @@ class TosaTestGen: return self.rng.random() elif dtype == DType.BOOL: return self.rng.choice([False, True]) + # TOSA specific INT4 weight range from -7 to 7 elif dtype == DType.INT4: - low, high = (-8, 8) + low, high = (-7, 8) elif dtype == DType.INT8: low, high = (-128, 128) elif dtype == DType.INT16: @@ -1977,6 +1979,7 @@ class TosaTestGen: TYPE_NARROW_INT_FP = [DType.INT8, DType.INT16, DType.FLOAT] TYPE_CONV2D = [ + [DType.INT8, DType.INT4, DType.INT32], [DType.INT8, DType.INT8, DType.INT32], [DType.INT16, DType.INT8, DType.INT48], DType.FLOAT, |