diff options
author | Louis Verhaard <louis.verhaard@arm.com> | 2020-12-16 12:02:28 +0100 |
---|---|---|
committer | Louis Verhaard <louis.verhaard@arm.com> | 2020-12-22 16:18:31 +0100 |
commit | 69b8480987fb078139e8b7d803cdfb5d76197651 (patch) | |
tree | 325c21fe43491d0b086b108cdbb03a8f38ed027c /ethosu | |
parent | f4e12beff027e8463aeaed7b75d5e48afb25c8b3 (diff) | |
download | ethos-u-vela-69b8480987fb078139e8b7d803cdfb5d76197651.tar.gz |
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>
Diffstat (limited to 'ethosu')
-rw-r--r-- | ethosu/vela/graph_optimiser.py | 1 |
1 files changed, 1 insertions, 0 deletions
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)}) |