Lines Matching defs:work
20 for (auto& work : works) { in waitWork() local
37 for (auto& work : works) { in waitFuture() local
67 c10::intrusive_ptr<::c10d::Work> work = pg->allreduce(tensors); in testAllreduce() local
102 c10::intrusive_ptr<::c10d::Work> work = pg->broadcast(tensors); in testBroadcast() local
128 c10::intrusive_ptr<::c10d::Work> work = pg->reduce(tensors); in testReduce() local
170 c10::intrusive_ptr<::c10d::Work> work = pg->allgather(outputs, tensors); in testAllgather() local
212 c10::intrusive_ptr<::c10d::Work> work = pg->gather(outputs, tensors); in testGather() local
262 c10::intrusive_ptr<::c10d::Work> work = pg->scatter(tensors, inputs); in testScatter() local
296 c10::intrusive_ptr<::c10d::Work> work = pg->send(sendTensors[i], 1, 0); in testSendRecv() local
304 c10::intrusive_ptr<::c10d::Work> work = pg->recv(recvTensors, 0, 0); in testSendRecv() local
307 c10::intrusive_ptr<::c10d::Work> work = in testSendRecv() local
321 for (const auto& work : works) { in testSendRecv() local