| 327 | net->add_before_backward(new NetCallback<Dtype>(run)); |
| 328 | } |
| 329 | void Net_after_backward(Net<Dtype>* net, bp::object run) { |
| 330 | net->add_after_backward(new NetCallback<Dtype>(run)); |
| 331 | } |
| 332 | |
| 333 | void Net_add_nccl(Net<Dtype>* net |
| 334 | #ifdef USE_NCCL |
nothing calls this directly
no test coverage detected