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