From 81942e9d59c1dcb1a9a54cb461f85bf582c7a3fd Mon Sep 17 00:00:00 2001 From: Jonas Ohlsson Date: Fri, 20 Aug 2021 09:33:28 +0200 Subject: 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 Change-Id: I4ceffcbb141af5ed50b0d1a9d1d67622e638c2a1 --- ethosu/vela/graph_optimiser_util.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'ethosu/vela/graph_optimiser_util.py') 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): -- cgit v1.2.1