Method main

tests/python/relax/test_transform_lambda_lift.py:69–80

(x1: R.Tensor((10, 5), "float32"), y1: R.Tensor((10, 5), "float32"))

    class Before:
        @R.function
        def main(x1: R.Tensor((10, 5), "float32"), y1: R.Tensor((10, 5), "float32")) -> R.Tensor(
            (10, 5), "float32"
        ):
            @R.function
            def inner(
                x2: R.Tensor((10, 5), "float32"), y2: R.Tensor((10, 5), "float32")
            ) -> R.Tensor((10, 5), "float32"):
                s: R.Tensor((10, 5), "float32") = R.add(x2, y2)
                return s

            gv1: R.Tensor((10, 5), "float32") = inner(x1, y1)
            return gv1

    before = Before
    expected = Expected