Function test

tests/python/relax/test_frontend_nn_op.py:43–50

(x: R.Tensor((1, 10), dtype="float32"), _io: R.Object)

    # fmt: off
    @R.function
    def test(x: R.Tensor((1, 10), dtype="float32"), _io: R.Object):
        R.func_attr({"num_input": 2})
        with R.dataflow():
            square: R.Tensor((1, 10), dtype="float32") = R.square(x)
            sqrt: R.Tensor((1, 10), dtype="float32") = R.sqrt(x)
            gv1 = (square, sqrt), (_io,)
            R.output(gv1)
        return gv1
    # fmt: on

    m = Model()