From fd0f802fe4d6102fad760125aa3d8e1176f2e397 Mon Sep 17 00:00:00 2001
From: Mirzam98 <mmrahorovic@hotmail.com>
Date: Fri, 7 May 2021 23:09:14 +0200
Subject: [PATCH] [fpgadataflow]: im2col node updated attributes

---
 src/finn/transformation/fpgadataflow/convert_to_hls_layers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py b/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
index a221b510a..b3a54ffbe 100644
--- a/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
+++ b/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
@@ -338,7 +338,7 @@ class InferPool_Batch(Transformation):
                     [im2col_in],
                     [im2col_out],
                     domain="finn.custom_op.general",
-                    stride=stride,
+                    stride=[stride, stride],
                     kernel_size=[k, k],
                     pad_amount=[pad, pad, pad, pad],
                     pad_value=pad_value,
-- 
GitLab