diff options
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): |