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

[Downsample] reflect changes to HLS kernel name change

now with 2D
parent 005aea52
No related branches found
No related tags found
No related merge requests found
......@@ -36,7 +36,7 @@ from finn.util.data_packing import npy_to_rtlsim_input, rtlsim_output_to_npy
class DownSampler(HLSCustomOp):
"""Corresponds to finn-hlslib ConvolutionInputGenerator_kernel1 function.
"""Corresponds to finn-hlslib ConvolutionInputGenerator_*_kernel1 function.
Basically performs a down sampling of the image removing rows and columns."""
def __init__(self, onnx_node):
......@@ -205,7 +205,7 @@ class DownSampler(HLSCustomOp):
def docompute(self):
self.code_gen_dict["$DOCOMPUTE$"] = [
"""ConvolutionInputGenerator_kernel1<IFMChannels, Input_precision,
"""ConvolutionInputGenerator_2D_kernel1<IFMChannels, Input_precision,
IFMDim, SIMD,Stride> (in0, out, numReps);"""
]
......
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