From 8602a2d624319d7749ed82fc7970edcd7f6785ae Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Wed, 20 Nov 2019 11:00:29 +0000
Subject: [PATCH] [Execution] Created StreamingFCLayer custom op and integrated
 it into code structure

---
 .../custom_op/fpgadataflow/streamingfclayer_batch.py   | 10 ++++++++++
 src/finn/custom_op/registry.py                         |  2 ++
 2 files changed, 12 insertions(+)
 create mode 100644 src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py

diff --git a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py
new file mode 100644
index 000000000..f7d1bb828
--- /dev/null
+++ b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py
@@ -0,0 +1,10 @@
+from finn.core.utils import get_by_name
+from finn.custom_op.fpgadataflow import HLSCustomOp
+
+
+class StreamingFCLayer_Batch(HLSCustomOp):
+    def make_shape_compatible_op(self, node):
+        pass
+
+    def infer_node_datatype(self, node, model):
+        pass
diff --git a/src/finn/custom_op/registry.py b/src/finn/custom_op/registry.py
index 7b1d2cf4f..38e7ef807 100644
--- a/src/finn/custom_op/registry.py
+++ b/src/finn/custom_op/registry.py
@@ -2,6 +2,7 @@
 # registered and plug in correctly into the infrastructure
 from finn.custom_op.fpgadataflow.streamingmaxpool import StreamingMaxPool
 from finn.custom_op.fpgadataflow.streamingmaxpool_batch import StreamingMaxPool_Batch
+from finn.custom_op.fpgadataflow.streamingfclayer_batch import StreamingFCLayer_Batch
 from finn.custom_op.multithreshold import MultiThreshold
 from finn.custom_op.xnorpopcount import XnorPopcountMatMul
 
@@ -12,3 +13,4 @@ custom_op["MultiThreshold"] = MultiThreshold
 custom_op["XnorPopcountMatMul"] = XnorPopcountMatMul
 custom_op["StreamingMaxPool"] = StreamingMaxPool
 custom_op["StreamingMaxPool_Batch"] = StreamingMaxPool_Batch
+custom_op["StreamingFCLayer_Batch"] = StreamingFCLayer_Batch
-- 
GitLab