From 13c6f6009a84ad83b2312e79170fb5017cfd4fe0 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Fri, 8 Nov 2019 08:39:41 -0500
Subject: [PATCH] [Code generation] Added StreamingFCLayer_Batch class to
layers.py and its init function
---
src/finn/backend/fpgadataflow/layers.py | 55 +++++++++++++++++++++++++
1 file changed, 55 insertions(+)
diff --git a/src/finn/backend/fpgadataflow/layers.py b/src/finn/backend/fpgadataflow/layers.py
index e69de29bb..d64a94bc3 100644
--- a/src/finn/backend/fpgadataflow/layers.py
+++ b/src/finn/backend/fpgadataflow/layers.py
@@ -0,0 +1,55 @@
+class StreamingFCLayer_Batch():
+ def __init__(self, node, model):
+ # Layer attributes
+ num_attr = len(node.attribute)
+ for k in range(num_attr):
+ if node.attribute[k].name == "PE":
+ self.PE = node.attribute[k].i
+ if node.attribute[k].name == "SIMD":
+ self.SIMD = node.attribute[k].i
+ if node.attribute[k].name == "MH":
+ self.MH = node.attribute[k].i
+ if node.attribute[k].name == "MW":
+ self.MW = node.attribute[k].i
+ if node.attribute[k].name == "resDataType":
+ self.resDataType = node.attribute[k].s.decode("utf-8")
+ if node.attribute[k].name == "resType":
+ self.resType = node.attribute[k].s.decode("utf-8")
+
+ # get other parameters
+ weights_shape = model.get_tensor_shape(node.input[1])
+ thresholds_shape = model.get_tensor_shape(node.input[2])
+ self.WMEM = weights_shape[2]
+ self.TMEM = thresholds_shape[0]
+ self.API = thresholds_shape[2]
+
+ def get_PE(self):
+ return self.PE
+
+ def get_SIMD(self):
+ return self.SIMD
+
+ def get_MH(self):
+ return self.MH
+
+ def get_MW(self):
+ return self.MW
+
+ def get_resDataType(self):
+ return self.resDataType
+
+ def get_resType(self):
+ return self.resType
+
+ def get_WMEM(self):
+ return self.WMEM
+
+ def get_TMEM(self):
+ return self.TMEM
+
+ def get_API(self):
+ return self.API
+
+
+
+
--
GitLab