commit 88d5b128fc86cecfa96aae09a7e7e9095f76b2a4
author Fredrik Svedberg <fredrik.svedberg@arm.com> Fri Sep 16 16:24:55 2022 +0200
committer Fredrik Svedberg <fredrik.svedberg@arm.com> Mon Sep 26 14:52:00 2022 +0000
tree 76de6d0918e465f93dfec1b2d3aa56a459f68bdf
parent 1cd39493163ea4cf74266b2957a5e54d1ed059bf

MLBEDSW-4075 PACK axis 0 + tanh fails with output diff

The test failed since the tanh had batch size > 1.
Added checks for batch size for all supported operators.

Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: I3570352740c40eb96bd9db965dfa3c91c81ff2ad

ethosu/vela/tosa_supported_operators.py

diff --git a/ethosu/vela/tosa_supported_operators.py b/ethosu/vela/tosa_supported_operators.py
index 15e1569..192862e 100644
--- a/ethosu/vela/tosa_supported_operators.py
+++ b/ethosu/vela/tosa_supported_operators.py
@@ -94,6 +94,9 @@
         self.generic_constraints.append(TosaSupportedOperators.constraint_rank)  # TODO not supported for all ops yet
         self.generic_constraints.append(TosaSupportedOperators.constraint_batch)  # TODO not supported for all ops yet
 
+        # Setup generic constraint exceptions
+        self.generic_constraints_exceptions = defaultdict(list)
+
         # Setup specific constraints. Note: the order matters
         self.specific_constraints = defaultdict(list)