(x: R.Tensor((2, 3), dtype="float32"), y: R.Tensor((2, 3), dtype="float32"))
| 47 | class Expected: |
| 48 | @R.function |
| 49 | def foo(x: R.Tensor((2, 3), dtype="float32"), y: R.Tensor((2, 3), dtype="float32")): |
| 50 | with R.dataflow(): |
| 51 | lv0 = R.add(x, y) |
| 52 | lv1 = lv0 |
| 53 | gv = R.multiply(lv0, lv0) |
| 54 | R.output(gv) |
| 55 | return gv |
| 56 | |
| 57 | verify(Before, Expected) |
| 58 |