Skip to content
Snippets Groups Projects
Commit 7ca9a9d4 authored by Tobi-Alonso's avatar Tobi-Alonso
Browse files

[HLSCustomOp] Add ChannelwiseOp_Batch for standalone functions with one parameter per channel

parent 21e62219
No related branches found
No related tags found
No related merge requests found
This diff is collapsed.
......@@ -49,6 +49,7 @@ from finn.custom_op.fpgadataflow.thresholding_batch import Thresholding_Batch
from finn.custom_op.fpgadataflow.addstreams_batch import AddStreams_Batch
from finn.custom_op.fpgadataflow.labelselect_batch import LabelSelect_Batch
from finn.custom_op.fpgadataflow.duplicatestreams_batch import DuplicateStreams_Batch
from finn.custom_op.fpgadataflow.channelwise_op_batch import ChannelwiseOp_Batch
# create a mapping of all known CustomOp names and classes
custom_op = {}
......@@ -70,6 +71,7 @@ custom_op["Thresholding_Batch"] = Thresholding_Batch
custom_op["AddStreams_Batch"] = AddStreams_Batch
custom_op["LabelSelect_Batch"] = LabelSelect_Batch
custom_op["DuplicateStreams_Batch"] = DuplicateStreams_Batch
custom_op["ChannelwiseOp_Batch"] = ChannelwiseOp_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