diff options
author | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2022-09-16 16:24:55 +0200 |
---|---|---|
committer | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2022-09-26 14:52:00 +0000 |
commit | 88d5b128fc86cecfa96aae09a7e7e9095f76b2a4 (patch) | |
tree | 76de6d0918e465f93dfec1b2d3aa56a459f68bdf /ethosu/vela/tosa_supported_operators.py | |
parent | 1cd39493163ea4cf74266b2957a5e54d1ed059bf (diff) | |
download | ethos-u-vela-88d5b128fc86cecfa96aae09a7e7e9095f76b2a4.tar.gz |
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
Diffstat (limited to 'ethosu/vela/tosa_supported_operators.py')
-rw-r--r-- | ethosu/vela/tosa_supported_operators.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/ethosu/vela/tosa_supported_operators.py b/ethosu/vela/tosa_supported_operators.py index 15e15695..192862ef 100644 --- a/ethosu/vela/tosa_supported_operators.py +++ b/ethosu/vela/tosa_supported_operators.py @@ -94,6 +94,9 @@ class TosaSupportedOperators: 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) |