Save Int16/UINT16 test outputs to native dtypes

* Int16/UInt16 reference outputs were previously saved to INT32
* Save those in their native dtypes and updated other affected code

Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: I0c3b7fba096a8cb1ddabef20ad13498b8f46d36f

2 files changed, 9 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
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 (