()
| 128 | |
| 129 | |
| 130 | def test_arange(): |
| 131 | @R.function |
| 132 | def foo(): |
| 133 | gv = R.arange(1, 10, 2) |
| 134 | return gv |
| 135 | |
| 136 | bb = relax.BlockBuilder() |
| 137 | with bb.function("foo", []): |
| 138 | gv = bb.emit(relax.op.arange(1, 10, 2)) |
| 139 | bb.emit_func_output(gv) |
| 140 | |
| 141 | _check(foo, bb.get()["foo"]) |
| 142 | |
| 143 | |
| 144 | def test_tril(): |