commit fc4bde92120567a98189f95cfe90bb1699d25809
author Jeremy Johnson <jeremy.johnson@arm.com> Thu Jan 25 12:53:21 2024 +0000
committer Eric Kunze <eric.kunze@arm.com> Tue Jan 30 16:11:26 2024 +0000
tree 48ab320c288440816734fc771a817257f3a229f9
parent 95a6710ffb8cadcb8658a967ab29cac1bffad930

Fix up shape operator test errors

Update serialization_lib to store SHAPE as INT64.

Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: Ie589cd6670dc79b77df981c81cd7c27b982f20fa

verif/generator/tosa_utils.py

diff --git a/verif/generator/tosa_utils.py b/verif/generator/tosa_utils.py
index 33db95f..6387d06 100644
--- a/verif/generator/tosa_utils.py
+++ b/verif/generator/tosa_utils.py
@@ -56,7 +56,7 @@
     """Types supported by the new data generation and compliance flow."""
     if isinstance(dtype, list) or isinstance(dtype, tuple):
         dtype = dtype[0]
-    return dtype in (DType.FP32, DType.FP16, DType.SHAPE)
+    return dtype in (DType.FP32, DType.FP16)
 
 
 def getOpNameFromOpListName(opName):