Skip to content
Snippets Groups Projects
Commit 8bbdf268 authored by auphelia's avatar auphelia
Browse files

[CustomOp] Add SameResize_Batch to registry

parent 2ca5c20e
No related branches found
No related tags found
No related merge requests found
......@@ -44,6 +44,7 @@ from finn.custom_op.fpgadataflow.streamingdatawidthconverter_batch import (
StreamingDataWidthConverter_Batch,
)
from finn.custom_op.fpgadataflow.globalaccpool_batch import GlobalAccPool_Batch
from finn.custom_op.fpgadataflow.sameresize_batch import SameResize_Batch
# create a mapping of all known CustomOp names and classes
custom_op = {}
......@@ -60,6 +61,7 @@ custom_op["MaxPoolNHWC"] = MaxPoolNHWC
custom_op["StreamingDataWidthConverter_Batch"] = StreamingDataWidthConverter_Batch
custom_op["StreamingFIFO"] = StreamingFIFO
custom_op["GlobalAccPool_Batch"] = GlobalAccPool_Batch
custom_op["SameResize_Batch"] = SameResize_Batch
def getCustomOp(node):
......
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