()
| 36 | """Simple testcase.""" |
| 37 | |
| 38 | def before(): |
| 39 | bb = relax.BlockBuilder() |
| 40 | x = relax.Var("x", R.Tensor([10, 20], "float32")) |
| 41 | with bb.function("main", [x]): |
| 42 | with bb.dataflow(): |
| 43 | lv0 = bb.emit_te(topi.add, x, relax.const(1, "float32")) |
| 44 | lv1 = bb.emit_te(topi.exp, lv0) |
| 45 | gv = bb.emit_output(bb.call_te(topi.squeeze, lv1)) |
| 46 | bb.emit_func_output(gv) |
| 47 | |
| 48 | return bb.get() |
| 49 | |
| 50 | def expected(): |
| 51 | bb = relax.BlockBuilder() |
no test coverage detected
searching dependent graphs…