From 0f773091c58e1480fcb7373ea104164e8126cedf Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Wed, 12 Feb 2020 23:43:42 +0100
Subject: [PATCH] [StreamingFC] implement i/o shape fxns
---
.../fpgadataflow/streamingfclayer_batch.py | 15 +++++++++++++--
1 file changed, 13 insertions(+), 2 deletions(-)
diff --git a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py
index ce21ad38c..be9b51e6a 100644
--- a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py
+++ b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py
@@ -208,10 +208,21 @@ class StreamingFCLayer_Batch(HLSCustomOp):
o_bits = self.get_output_datatype().bitwidth()
return o_bits * self.get_nodeattr("PE")
- def get_number_output_values(self):
+ def get_folded_input_shape(self):
+ mw = self.get_nodeattr("MW")
+ simd = self.get_nodeattr("SIMD")
+ sf = mw // simd
+ return (1, sf, simd)
+
+ def get_folded_output_shape(self):
mh = self.get_nodeattr("MH")
pe = self.get_nodeattr("PE")
- return mh // pe
+ nf = mh // pe
+ return (1, nf, pe)
+
+ def get_number_output_values(self):
+ nf = self.get_folded_output_shape()[1]
+ return nf
def get_template_param_values(self):
ret = dict()
--
GitLab