Skip to content
Snippets Groups Projects
Commit 4e90bb97 authored by auphelia's avatar auphelia
Browse files

[customOp class] changed verify_node() to abstract method

parent aebdefc0
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):
......@@ -80,6 +79,9 @@ class CustomOp(ABC):
ONNX graph."""
pass
@abstractmethod
def verify_node(self):
inst = CustomOp_Construct[self.onnx_node.op_type]
inst.verify_construct(self.onnx_node)
"""Verifies that all attributes the node needs are there and
that particular attributes are set correctly. Also checks if
the number of inputs is equal to the expected number"""
pass
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