(data, ignore_idx=[])
| 91 | return tvm.runtime.tensor(data) |
| 92 | |
| 93 | def _tvm_to_numpy(data, ignore_idx=[]): |
| 94 | if isinstance(data, tvm_ffi.Array): |
| 95 | return [_tvm_to_numpy(d) for i, d in enumerate(data) if i not in ignore_idx] |
| 96 | if isinstance(data, tvm.runtime.Tensor): |
| 97 | return data.numpy() |
| 98 | return data |
| 99 | |
| 100 | def _gen_weights(out_sinfo): |
| 101 | if isinstance(out_sinfo, TupleStructInfo): |
no test coverage detected
searching dependent graphs…