From 7935972b1ae427b597ce2e817c5071c44d7ba56e Mon Sep 17 00:00:00 2001
From: Tai Ly <tai.ly@arm.com>
Date: Thu, 31 Aug 2023 18:17:38 +0000
Subject: [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 | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 @@ class OutputShaper:
 
     @staticmethod
     def dimOp(ser, rng, a, axis, error_name=None):
-        output_shape = [1]
+        output_shape = []
 
         if error_name == ErrorIf.WrongOutputType:
             all_dtypes = [
-- 
cgit v1.2.1