Skip to content
Snippets Groups Projects
Commit 28321ccc authored by auphelia's avatar auphelia
Browse files

[CustomOp init] Added verify functionality to customOp class

parent a8d1b5c1
No related branches found
No related tags found
No related merge requests found
from abc import ABC, abstractmethod
from finn.core.utils import get_by_name
import onnx.helper as helper
from finn.custom_op.verify_custom_op_construct import CustomOp_Construct
class CustomOp(ABC):
......@@ -78,3 +79,7 @@ class CustomOp(ABC):
"""Execute this CustomOp instance, given the execution context and
ONNX graph."""
pass
def verify_node(self):
inst = CustomOp_Construct[self.onnx_node.op_type]
inst.verify_construct(self.onnx_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