Skip to content
Snippets Groups Projects
Commit f952b76b authored by auphelia's avatar auphelia
Browse files

[Execution] Adjust im2col and maxpool hls nodes to new HLSCustomOp functions

parent 9dba93d4
No related branches found
No related tags found
No related merge requests found
......@@ -82,7 +82,7 @@ class ConvolutionInputGenerator(HLSCustomOp):
def global_includes(self):
self.code_gen_dict["$GLOBALS$"] = ['#include "slidingwindow.h"']
def defines(self):
def defines(self, var):
numReps = 1
self.code_gen_dict["$DEFINES$"] = [
"""#define ConvKernelDim1 {}\n #define IFMChannels1 {}
......
......@@ -72,7 +72,7 @@ class StreamingMaxPool_Batch(HLSCustomOp):
def global_includes(self):
self.code_gen_dict["$GLOBALS$"] = ['#include "maxpool.h"']
def defines(self):
def defines(self, var):
numReps = 2
self.code_gen_dict["$DEFINES$"] = [
"""#define ImgDim {}\n #define PoolDim {}\n
......@@ -197,3 +197,9 @@ class StreamingMaxPool_Batch(HLSCustomOp):
int(self.get_nodeattr("ImgDim") / self.get_nodeattr("PoolDim")),
)
]
def blackboxfunction(self):
pass
def pragmas(self):
pass
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