diff --git a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py index 70e3b3c9503789de7fe0c8c803a5a40980bbd126..478c22b526d4836856d865179862b94595cb9974 100644 --- a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py +++ b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py @@ -208,13 +208,15 @@ class StreamingFCLayer_Batch(HLSCustomOp): export_odt = DataType.BINARY odt_hls = export_odt.get_hls_datatype_str() f_thresh.write( - "static ThresholdsActivation<{},{},{},{},{},{}> threshs = ".format( + "static ThresholdsActivation<{},{},{},{},{},{},{}> threshs \ + = ".format( self.get_nodeattr("TMEM"), self.get_nodeattr("PE"), threshold_tensor.shape[-1], tdt_hls, odt_hls, self.get_nodeattr("ActVal"), + "std::less_equal<%s>" % tdt_hls, ) ) f_thresh.write(thresholds_hls_code)