diff options
author | Jonas Ohlsson <jonas.ohlsson@arm.com> | 2021-08-20 09:33:28 +0200 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2021-08-23 07:04:51 +0000 |
commit | 81942e9d59c1dcb1a9a54cb461f85bf582c7a3fd (patch) | |
tree | 8ecaa13f3419ffe86acad86cc0482bec92872416 /ethosu/vela/graph_optimiser_util.py | |
parent | cc34d5d33938b6cbd9f8b11883f12e8c3097b613 (diff) | |
download | ethos-u-vela-81942e9d59c1dcb1a9a54cb461f85bf582c7a3fd.tar.gz |
MLBEDSW-4913 Fix inception_v1/v3 output diff
Fix inception_v1/v3 output diffs.
Removing the Squeeze operator in the graph optimisation step.
The squeeze operator removes dimensions of size 1 from tensor shape.
The memory layout is preserved.
Signed-off-by: Jonas Ohlsson <jonas.ohlsson@arm.com>
Change-Id: I4ceffcbb141af5ed50b0d1a9d1d67622e638c2a1
Diffstat (limited to 'ethosu/vela/graph_optimiser_util.py')
-rw-r--r-- | ethosu/vela/graph_optimiser_util.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/ethosu/vela/graph_optimiser_util.py b/ethosu/vela/graph_optimiser_util.py index 0b44b8f6..5e676f18 100644 --- a/ethosu/vela/graph_optimiser_util.py +++ b/ethosu/vela/graph_optimiser_util.py @@ -23,7 +23,10 @@ from .shape4d import Shape4D from .tensor import check_quantized_tens_scaling_equal -memory_only_ops = (Op.Reshape,) +memory_only_ops = ( + Op.Reshape, + Op.Squeeze, +) def _avoid_nhcwb16_for_concat(tens): |