| 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); }; |
| 158 | return m_network->graph->compile({{m_out_var, std::move(cb)}}); |
| 159 | } |
| 160 | }; |
| 161 | |
| 162 | namespace { |
no test coverage detected