Searched defs:allreduce_sparse (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/csrc/distributed/c10d/ | ||
| D | FakeProcessGroup.hpp | 36 c10::intrusive_ptr<Work> allreduce_sparse( in allreduce_sparse() function in c10d::FakeProcessGroup |
| D | Backend.hpp | 98 virtual c10::intrusive_ptr<Work> allreduce_sparse( in allreduce_sparse() function in c10d::Backend |
| D | ProcessGroupNCCL.cpp | 3337 c10::intrusive_ptr<Work> ProcessGroupNCCL::allreduce_sparse( in checkForNCCLErrorsInternal() function in c10d::ProcessGroupNCCL |
| D | ProcessGroupGloo.cpp | 1548 c10::intrusive_ptr<Work> ProcessGroupGloo::allreduce_sparse( in allreduce_sparse() function in c10d::ProcessGroupGloo |