diff options
Diffstat (limited to 'ethosu/vela/operation.py')
-rw-r--r-- | ethosu/vela/operation.py | 3 |
1 files changed, 3 insertions, 0 deletions
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 |