Function _expected

tests/python/relax/test_tvmscript_parser.py:438–447

(name: str)

            return gv0

    def _expected(name: str):
        n, m = tirx.Var("n", "int64"), tirx.Var("m", "int64")
        x = relax.Var("x", R.Tensor([m, n], "float32"))
        bb = relax.BlockBuilder()
        with bb.function(name, (x,)):
            out = bb.emit(
                relax.call_dps_packed("extern_func", x, R.Tensor((m, n), dtype="float32"))
            )
            bb.emit_func_output(out)
        return bb.get()[name]

    _check(foo, _expected("foo"))
    _check(bar, _expected("bar"))