From cf7da10987cac3fc68cf180a9af665fe06d608fa Mon Sep 17 00:00:00 2001
From: Jacob Bohlin <jacob.bohlin@arm.com>
Date: Wed, 20 May 2020 09:03:40 +0200
Subject: MLBEDSW-1716: Transpose Convolution support

Change-Id: Ie6d8d6de9f3447f19ba06aafa9fa480fc96a973b
Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
---
 ethosu/vela/operation.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'ethosu/vela/operation.py')

diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py
index e8a03b7d..51311ef7 100644
--- a/ethosu/vela/operation.py
+++ b/ethosu/vela/operation.py
@@ -75,6 +75,9 @@ input and output tensors, as well as an attribute dictionary."""
                 if len(self.inputs) >= 3:
                     bias_idx = 2
 
+            elif self.type == "Conv2DBackpropInputSwitchedBias":
+                bias_idx = 3
+
         elif npu_block_type == NpuBlockType.Pooling:
             ifm_idx = 0
             ofm_idx = 0
-- 
cgit v1.2.1