diff options
Diffstat (limited to 'verif/generator/tosa_test_gen.py')
-rw-r--r-- | verif/generator/tosa_test_gen.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index bc931dc..8440853 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -195,6 +195,10 @@ class TosaTestGen: return np.int8(self.rng.integers(low=low, high=high, size=shape)) elif dtype == DType.UINT8: return np.uint8(self.rng.integers(low=low, high=high, size=shape)) + elif dtype == DType.INT16: + return np.int16(self.rng.integers(low=low, high=high, size=shape)) + elif dtype == DType.UINT16: + return np.uint16(self.rng.integers(low=low, high=high, size=shape)) elif dtype in (DType.INT48, DType.SHAPE): return np.int64(self.rng.integers(low=low, high=high, size=shape)) elif dtype in ( |