From c5f928bde85f34981308ffb8d513839b034da7d2 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Wed, 27 May 2020 17:46:25 +0100
Subject: [PATCH] [CustomOp] Add new node attribute "dw" to Im2Col
---
src/finn/custom_op/im2col.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/src/finn/custom_op/im2col.py b/src/finn/custom_op/im2col.py
index 16446c15d..b0d94b158 100644
--- a/src/finn/custom_op/im2col.py
+++ b/src/finn/custom_op/im2col.py
@@ -79,6 +79,7 @@ class Im2Col(CustomOp):
"input_shape": ("s", True, ""),
"pad_amount": ("i", False, 0),
"pad_value": ("i", False, 0),
+ "dw": ("i", False, 0),
}
def make_shape_compatible_op(self, model):
--
GitLab