commit 81942e9d59c1dcb1a9a54cb461f85bf582c7a3fd
author Jonas Ohlsson <jonas.ohlsson@arm.com> Fri Aug 20 09:33:28 2021 +0200
committer patrik.gustavsson <patrik.gustavsson@arm.com> Mon Aug 23 07:04:51 2021 +0000
tree 8ecaa13f3419ffe86acad86cc0482bec92872416
parent cc34d5d33938b6cbd9f8b11883f12e8c3097b613

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

ethosu/vela/graph_optimiser_util.py

diff --git a/ethosu/vela/graph_optimiser_util.py b/ethosu/vela/graph_optimiser_util.py
index 0b44b8f..5e676f1 100644
--- a/ethosu/vela/graph_optimiser_util.py
+++ b/ethosu/vela/graph_optimiser_util.py
@@ -23,7 +23,10 @@
 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):