diff --git a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py index 80109d0fae20e86d520d6769f7c5ace6911d611d..4857620a5d9b2f7c035f2da40c1d71a884ae1178 100644 --- a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py +++ b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py @@ -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"