()
| 290 | |
| 291 | @requires_coreml_runtime |
| 292 | def test_subgraph1(): |
| 293 | x = relax.Var("x", relax.TensorStructInfo([10, 10], "float32")) |
| 294 | y = relax.Var("y", relax.TensorStructInfo([10, 10], "float32")) |
| 295 | bb = relax.BlockBuilder() |
| 296 | with bb.function("main", [x, y]): |
| 297 | with bb.dataflow(): |
| 298 | lv0 = bb.emit(relax.op.multiply(x, y)) |
| 299 | lv1 = bb.emit(relax.op.nn.softmax(lv0)) |
| 300 | gv = bb.emit_output(lv1) |
| 301 | bb.emit_func_output(gv) |
| 302 | mod = bb.get() |
| 303 | x_data = tvm.runtime.tensor(np.random.rand(10, 10).astype("float32"), dev) |
| 304 | y_data = tvm.runtime.tensor(np.random.rand(10, 10).astype("float32"), dev) |
| 305 | verify(mod, [x_data, y_data]) |
| 306 | |
| 307 | |
| 308 | @requires_coreml_runtime |
nothing calls this directly
no test coverage detected
searching dependent graphs…