()
| 83 | |
| 84 | |
| 85 | def test_ones_like(): |
| 86 | @R.function |
| 87 | def foo(x: R.Tensor((2, 3), "float32")) -> R.Tensor((2, 3), "float32"): |
| 88 | gv: R.Tensor((2, 3), "float32") = R.ones_like(x) |
| 89 | return gv |
| 90 | |
| 91 | x = relax.Var("x", R.Tensor((2, 3), "float32")) |
| 92 | bb = relax.BlockBuilder() |
| 93 | with bb.function("foo", [x]): |
| 94 | gv = bb.emit(relax.op.ones_like(x)) |
| 95 | bb.emit_func_output(gv) |
| 96 | |
| 97 | _check(foo, bb.get()["foo"]) |
| 98 | |
| 99 | |
| 100 | def test_zeros(): |