(x: R.Tensor((128, 128), "float32"))
| 206 | |
| 207 | @R.function |
| 208 | def foo(x: R.Tensor((128, 128), "float32")) -> R.Tensor((128, 128), "float32"): |
| 209 | cls = TestModule |
| 210 | gv0 = R.call_tir(cls.tir_func, x, R.Tensor((128, 128), dtype="float32")) |
| 211 | return gv0 |
| 212 | |
| 213 | x = relax.Var("x", R.Tensor((128, 128), "float32")) |
| 214 | bb = relax.BlockBuilder() |