From 52ff078a1ab08c5cc89c788d10479b855f442653 Mon Sep 17 00:00:00 2001 From: Yaman Umuroglu <maltanar@gmail.com> Date: Mon, 5 Oct 2020 16:51:44 +0200 Subject: [PATCH] [Refactor] remove duplicate getCustomOp, should come from finn-base --- src/finn/custom_op/__init__.py | 12 ------------ 1 file changed, 12 deletions(-) diff --git a/src/finn/custom_op/__init__.py b/src/finn/custom_op/__init__.py index 08207a351..06fc7e565 100644 --- a/src/finn/custom_op/__init__.py +++ b/src/finn/custom_op/__init__.py @@ -76,15 +76,3 @@ custom_op["DuplicateStreams_Batch"] = DuplicateStreams_Batch custom_op["Vector_Vector_Activate_Batch"] = Vector_Vector_Activate_Batch custom_op["ChannelwiseOp_Batch"] = ChannelwiseOp_Batch custom_op["IODMA"] = IODMA - - -def getCustomOp(node): - "Return a FINN CustomOp instance for the given ONNX node, if it exists." - op_type = node.op_type - try: - # lookup op_type in registry of CustomOps - inst = custom_op[op_type](node) - return inst - except KeyError: - # exception if op_type is not supported - raise Exception("Custom op_type %s is currently not supported." % op_type) -- GitLab