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

[Conv] check for dilation=1 differently

parent 82a1a1e0
No related branches found
No related tags found
No related merge requests found
......@@ -68,7 +68,7 @@ class ConvolutionInputGenerator(HLSCustomOp):
"SIMD": ("i", True, 0),
"Stride": ("ints", True, [1, 1]), # [H, W] = [Y, X]
# note: only dilation=1 supported for now
"Dilation": ("ints", True, [1, 1], {[1, 1]}), # [H, W] = [Y, X]
"Dilation": ("ints", True, [1, 1]), # [H, W] = [Y, X]
# FINN DataTypes for inputs, weights, outputs
"inputDataType": ("s", True, ""),
"outputDataType": ("s", True, ""),
......@@ -97,6 +97,8 @@ class ConvolutionInputGenerator(HLSCustomOp):
if name in props_to_check:
is_square = ret[0] == ret[1]
assert is_square, "Only square %s supported" % name
if name == "Dilation":
assert ret[0] == ret[1] == 1, "Only dilation=1 supported"
return ret
def get_normal_input_shape(self):
......
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