From 169dc12f87ad7707d53a28dd4d29211ba9f9aaf7 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Tue, 21 Jul 2020 10:49:59 +0100
Subject: [PATCH] [CustomOp] Add get_exp_cycles for downsampler
---
src/finn/custom_op/fpgadataflow/downsampler.py | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/src/finn/custom_op/fpgadataflow/downsampler.py b/src/finn/custom_op/fpgadataflow/downsampler.py
index 0ce4379a2..4acca3758 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")
--
GitLab