| 150 | } |
| 151 | |
| 152 | std::unique_ptr<cg::AsyncExecutable> compile_without_copy() { |
| 153 | return m_network->graph->compile({{m_out_var, nullptr}}); |
| 154 | } |
| 155 | |
| 156 | std::unique_ptr<cg::AsyncExecutable> compile_with_copy(HostTensorND& host) { |
| 157 | auto cb = [&host](const DeviceTensorND& dv) mutable { host.copy_from(dv); }; |
no test coverage detected