diff --git a/src/finn/custom_op/fpgadataflow/pool_batch.py b/src/finn/custom_op/fpgadataflow/pool_batch.py index 3b0f3fbf00072659b9f43047aab3a26f7abf8f3a..4a2fa6889ae0ebb94976d50b0fc8362d01a63bea 100644 --- a/src/finn/custom_op/fpgadataflow/pool_batch.py +++ b/src/finn/custom_op/fpgadataflow/pool_batch.py @@ -143,7 +143,7 @@ class Pool_Batch(HLSCustomOp): k = self.get_nodeattr("KernelSize") odim = self.get_nodeattr("OutImgDim") batch_size = self.get_nodeattr("BatchSize") - exp_cycles = (ifm_ch * k * k) / pe * odim * odim * batch_size + exp_cycles = ((ifm_ch * k * k) / pe) * odim * odim * batch_size return int(exp_cycles) def get_instream_width(self):