[CustomOp] Add get_exp_cycles for pool_batch and fmpadding

src/finn/custom_op/fpgadataflow/fmpadding_batch.py

@@ -42,6 +42,14 @@ class FMPadding_Batch(HLSCustomOp):
         pad = self.get_nodeattr("Padding")
         return idim + pad
 
+    def get_exp_cycles(self):
+        odim = self.get_padded_odim()
+        channels = self.get_nodeattr("NumChannels")
+        simd = self.get_nodeattr("SIMD")
+        batch_size = self.get_nodeattr("numInputVectors")
+        exp_cycles = channels / simd * batch_size * odim * odim
+        return exp_cycles
+
     def get_normal_input_shape(self):
         idim = self.get_nodeattr("ImgDim")
         num_ch = self.get_nodeattr("NumChannels")

src/finn/custom_op/fpgadataflow/pool_batch.py

@@ -136,6 +136,10 @@ class Pool_Batch(HLSCustomOp):
         folded_oshape = self.get_folded_output_shape()
         return np.prod(folded_oshape[1:-1])
 
+    def get_exp_cycles(self):
+        # Channels/PE * 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()
         pe = self.get_nodeattr("PE")