(mod, enable_nnapi)
| 73 | |
| 74 | |
| 75 | def _build(mod, enable_nnapi): |
| 76 | if isinstance(mod, tvm.relax.expr.Call): |
| 77 | mod = tvm.IRModule.from_expr(mod) |
| 78 | |
| 79 | if enable_nnapi: |
| 80 | mod = tvm.relax.transform.FoldConstant()(mod) |
| 81 | mod = reshape_matmul(mod) |
| 82 | mod = decompose_clip(mod) |
| 83 | mod = partition_for_nnapi(mod) |
| 84 | |
| 85 | mod = tvm.relax.transform.RunCodegen()(mod) |
| 86 | ex = tvm.compile(mod, target={"kind": "llvm", "mtriple": "aarch64-linux-android"}) |
| 87 | |
| 88 | return ex |
| 89 | |
| 90 | |
| 91 | def _run(remote, tracker, ex, inputs): |
no test coverage detected
searching dependent graphs…