Skip to content
Snippets Groups Projects
Commit 8602a2d6 authored by auphelia's avatar auphelia
Browse files

[Execution] Created StreamingFCLayer custom op and integrated it into code structure

parent 27345649
No related branches found
No related tags found
No related merge requests found
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
......@@ -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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment