Method main

tests/python/relax/test_transform_convert_dataflow.py:86–92

(x: R.Tensor, y: R.Tensor)

    class Expected:
        @R.function
        def main(x: R.Tensor, y: R.Tensor) -> R.Tensor:
            with R.dataflow():
                z = R.add(x, y)
                w = R.multiply(z, y)
                v = R.add(w, x)
                R.output(v)
            return v

    After = relax.transform.ConvertToDataflow()(Before)
    tvm.ir.assert_structural_equal(After, Expected)