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

[StreamingFC] fix codegen for ibp w-nonbp

parent 1a0ae6a8
No related branches found
No related tags found
No related merge requests found
......@@ -56,7 +56,6 @@ class StreamingFCLayer_Batch(HLSCustomOp):
def get_template_param_values(self):
ret = dict()
inp_hls_str = self.get_input_datatype().get_hls_datatype_str()
wt_hls_str = self.get_weight_datatype().get_hls_datatype_str()
out_hls_str = self.get_output_datatype().get_hls_datatype_str()
inp_is_binary = self.get_input_datatype() == DataType.BINARY
out_is_binary = self.get_output_datatype() == DataType.BINARY
......@@ -77,7 +76,7 @@ class StreamingFCLayer_Batch(HLSCustomOp):
ret["TWeightI"] = "Recast<Binary>"
elif inp_is_bipolar and (not wt_is_bipolar):
ret["TSrcI"] = "Recast<Binary>"
ret["TWeightI"] = "Slice<%s>" % wt_hls_str
ret["TWeightI"] = "Identity"
elif (not inp_is_bipolar) and (not wt_is_bipolar):
ret["TSrcI"] = "Slice<%s>" % inp_hls_str
ret["TWeightI"] = "Identity"
......
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