diff --git a/src/finn/custom_op/fpgadataflow/globalaccpool_batch.py b/src/finn/custom_op/fpgadataflow/globalaccpool_batch.py index 477fcf50720dc45b9e323155f1e94ec2451d3877..7bfd4eb59b6e740d9d39ae407948926bac848b0b 100644 --- a/src/finn/custom_op/fpgadataflow/globalaccpool_batch.py +++ b/src/finn/custom_op/fpgadataflow/globalaccpool_batch.py @@ -183,8 +183,8 @@ class GlobalAccPool_Batch(HLSCustomOp): return np.prod(self.get_folded_output_shape()[1:-1]) def get_exp_cycles(self): - # Channels/PE * batch size (ofmdim = 1) - return np.prod(self.get_folded_output_shape()[:-1]) + # Channels/PE * batch size * idim * idim + return np.prod(self.get_folded_input_shape()[:-1]) def execute_node(self, context, graph): mode = self.get_nodeattr("exec_mode")