(sess: di.Session, np_array: np.array, device)
| 136 | |
| 137 | |
| 138 | def _numpy_to_worker_0(sess: di.Session, np_array: np.array, device): |
| 139 | x_array = sess.empty(np_array.shape, "float32", device=device) |
| 140 | host_array = tvm.runtime.tensor(np_array, device=device) |
| 141 | sess.copy_to_worker_0(host_array, x_array) |
| 142 | return x_array |
| 143 | |
| 144 | |
| 145 | def _numpy_from_worker_0(sess: di.Session, remote_array, shape, dtype): |
no test coverage detected
searching dependent graphs…