From 69b8480987fb078139e8b7d803cdfb5d76197651 Mon Sep 17 00:00:00 2001 From: Louis Verhaard Date: Wed, 16 Dec 2020 12:02:28 +0100 Subject: MLBEDSW-3495: Tensor resampling mode for transpose Sets IFM's resampling mode for transpose convolutions. Change-Id: I11744a932aea7c11fa70036c43a7ed01ea4b2929 Signed-off-by: Louis Verhaard --- ethosu/vela/graph_optimiser.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py index 00edf835..e468f6df 100644 --- a/ethosu/vela/graph_optimiser.py +++ b/ethosu/vela/graph_optimiser.py @@ -209,6 +209,7 @@ def fixup_conv2d_backprop(op, arch, nng): 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)}) -- cgit v1.2.1