commit 69b8480987fb078139e8b7d803cdfb5d76197651
author Louis Verhaard <louis.verhaard@arm.com> Wed Dec 16 12:02:28 2020 +0100
committer Louis Verhaard <louis.verhaard@arm.com> Tue Dec 22 16:18:31 2020 +0100
tree 325c21fe43491d0b086b108cdbb03a8f38ed027c
parent f4e12beff027e8463aeaed7b75d5e48afb25c8b3

MLBEDSW-3495: Tensor resampling mode for transpose

Sets IFM's resampling mode for transpose convolutions.

Change-Id: I11744a932aea7c11fa70036c43a7ed01ea4b2929
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>

ethosu/vela/graph_optimiser.py

diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py
index 00edf83..e468f6d 100644
--- a/ethosu/vela/graph_optimiser.py
+++ b/ethosu/vela/graph_optimiser.py
@@ -209,6 +209,7 @@
         op.inputs[0], op.inputs[2] = op.inputs[2], op.inputs[0]
         op.set_ifm_ofm_shapes()
         op.type = Op.Conv2DBackpropInputSwitchedBias
+        op.ifm.resampling_mode = resampling_mode.TRANSPOSE
 
         # Update strides
         op.attrs.update({"stride_w": 1, "stride_h": 1, "strides": (1, 1, 1, 1)})