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

[Test] Added test function to test verification of MultiThreshold node

parent 7ed9093d
No related branches found
No related tags found
No related merge requests found
import onnx
from finn.custom_op.layout_custom_ops import CustomOp_Layout
def test_verify_layout_custom_ops():
m_node = onnx.helper.make_node(
"MultiThreshold",
["v", "thresholds"],
["out"],
domain="finn",
out_scale=2.0,
out_bias=-1.0,
out_dtype="",
)
inst = CustomOp_Layout[m_node.op_type]
inst.verify_layout(m_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