Home
last modified time | relevance | path

Searched defs:allreduce_sparse (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DFakeProcessGroup.hpp36 c10::intrusive_ptr<Work> allreduce_sparse( in allreduce_sparse() function in c10d::FakeProcessGroup
H A DBackend.hpp98 virtual c10::intrusive_ptr<Work> allreduce_sparse( in allreduce_sparse() function in c10d::Backend
H A DProcessGroupNCCL.cpp3337 c10::intrusive_ptr<Work> ProcessGroupNCCL::allreduce_sparse( in checkForNCCLErrorsInternal() function in c10d::ProcessGroupNCCL
H A DProcessGroupGloo.cpp1548 c10::intrusive_ptr<Work> ProcessGroupGloo::allreduce_sparse( in allreduce_sparse() function in c10d::ProcessGroupGloo