diff --git a/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py b/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py index 2344e12f7e87634c189563f9cde7b1c861a3606e..266870a1bc8964f7ca56d2effd897a9e5fafc53e 100644 --- a/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py +++ b/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py @@ -95,6 +95,10 @@ class StreamingMaxPool_Batch(HLSCustomOp): folded_oshape = self.get_folded_output_shape() return np.prod(folded_oshape[:-1]) + def get_exp_cycles(self): + # Channels * batch size * odim * odim + return np.prod(self.get_folded_output_shape()[:-1]) + def get_instream_width(self): dt_bits = self.get_input_datatype().bitwidth() ifm_ch = self.get_nodeattr("NumChannels") diff --git a/src/finn/custom_op/fpgadataflow/thresholding_batch.py b/src/finn/custom_op/fpgadataflow/thresholding_batch.py index fa33c70218fab16f106da45e296f0d59ae4ea606..c2e3739e8f62b5ce0459ee8fbb1f3dcda7b50c1e 100644 --- a/src/finn/custom_op/fpgadataflow/thresholding_batch.py +++ b/src/finn/custom_op/fpgadataflow/thresholding_batch.py @@ -215,6 +215,10 @@ class Thresholding_Batch(HLSCustomOp): nf = np.prod(self.get_folded_output_shape()[:-1]) return nf + def get_exp_cycles(self): + # Channels/PE * batch size * fmdim * fmdim + return np.prod(self.get_folded_output_shape()[:-1]) + def get_template_param_values(self): """Returns the template parameter values according to input, output and weight data types."""