Function _check

tests/python/relax/test_tvmscript_parser_op_image.py:26–34

(
    parsed: relax.Function | IRModule,
    expect: relax.Function | IRModule | None,
)

def _check(
    parsed: relax.Function | IRModule,
    expect: relax.Function | IRModule | None,
):
    test = parsed.script(show_meta=True)
    roundtrip_mod = tvm.script.from_source(test)
    tvm.ir.assert_structural_equal(parsed, roundtrip_mod)
    if expect:
        tvm.ir.assert_structural_equal(parsed, expect)


def test_resize2d():