commit 7935972b1ae427b597ce2e817c5071c44d7ba56e
author Tai Ly <tai.ly@arm.com> Thu Aug 31 18:17:38 2023 +0000
committer Tai Ly <tai.ly@arm.com> Fri Sep 01 15:56:01 2023 +0000
tree df61ade5f69d1e6eb51e2eea9921e6f09e92978b
parent 0633c3a2603ea626ec7cc035dc8f6dc7f9f07795

[reference model] Fix dim test generation

In tosa_test_gen.py, change output_shape for dim operators to [] instead of [1]

Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: I8e7901e02831e9a467bce17e55e9379cb5f1b739

verif/generator/tosa_test_gen.py

diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py
index 8c18e67..3014c81 100644
--- a/verif/generator/tosa_test_gen.py
+++ b/verif/generator/tosa_test_gen.py
@@ -4732,7 +4732,7 @@
 
     @staticmethod
     def dimOp(ser, rng, a, axis, error_name=None):
-        output_shape = [1]
+        output_shape = []
 
         if error_name == ErrorIf.WrongOutputType:
             all_dtypes = [