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