MLBEDSW-2420: Improved support for dilated convolution
- Dilation added to SET_KERNEL_STRIDE instruction
- Kernel height/width adjusted for dilation
- Updated padding calculation
- Updated weight compression
Change-Id: I0c8190223e223b039a305aba0f37896ae1de2b80
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py
index 338f962..e8a03b7 100644
--- a/ethosu/vela/operation.py
+++ b/ethosu/vela/operation.py
@@ -194,6 +194,10 @@
return inputs, axis
+ def get_dilation_h_w(self):
+ _, dilation_h, dilation_w, _ = self.attrs.get("dilation", (1, 1, 1, 1))
+ return dilation_h, dilation_w
+
split_ops = set(("Split", "SplitV", "StridedSlice", "Slice", "UnpackReshaped"))
def is_split_op(self):