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 0000000000000000000000000000000000000000..f7d1bb828dd3af255937ec406bf18cfcc3d5f7d9 --- /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 7b1d2cf4f405552f13ca5d1c9b21a858f350130a..38e7ef8077e8e43adfcac164071d1f7e82145644 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