Skip to content
Snippets Groups Projects
Commit 2d5d5fd2 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[ConvertToHLS] make 1D SWG kernel check more constrained

parent e7ca8464
No related branches found
No related tags found
No related merge requests found
......@@ -213,10 +213,11 @@ class InferConvInpGen(Transformation):
1D depthwise separable convolutions"""
% n.name
)
if stride_h > 1 or stride_w > 1:
if (stride_h > 1 or stride_w > 1) and (not depthwise):
assert (
stride_h < k_h and stride_w < k_w
), """%s: Stride value must be smaller than kernel dim"""
), """%s: Stride value must be smaller than kernel dim
for non-depthwise (dense) convolutions"""
ConvInpGen_node = helper.make_node(
"ConvolutionInputGenerator1D",
[ConvInpGen_input],
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment