diff --git a/src/finn/custom_op/fpgadataflow/downsampler.py b/src/finn/custom_op/fpgadataflow/downsampler.py index 0ce4379a2c41baa5bc009e9df7623d133ee89a09..4acca3758661047ea43827c5d79a0727f13302b5 100644 --- a/src/finn/custom_op/fpgadataflow/downsampler.py +++ b/src/finn/custom_op/fpgadataflow/downsampler.py @@ -36,6 +36,14 @@ class DownSampler(HLSCustomOp): stride = self.get_nodeattr("Stride") return int(np.floor((idim - 1) / stride) + 1) + def get_exp_cycles(self): + fmdim = self.get_downsampled_odim() + channels = self.get_nodeattr("NumChannels") + simd = self.get_nodeattr("SIMD") + batch_size = self.get_nodeattr("numInputVectors") + exp_cycles = channels / simd * batch_size * fmdim * fmdim + return exp_cycles + def get_normal_input_shape(self): idim = self.get_nodeattr("ImgDim") num_ch = self.get_nodeattr("NumChannels")