Home
last modified time | relevance | path

Searched defs:allreduce_coalesced (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dcpp_c10d_extension.cpp38 c10::intrusive_ptr<Work> ProcessGroupTest::allreduce_coalesced( in allreduce_coalesced() function in c10d::ProcessGroupTest
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DPyProcessGroup.hpp106 c10::intrusive_ptr<Work> allreduce_coalesced( in allreduce_coalesced() function in c10d::PyProcessGroup
H A DFakeProcessGroup.hpp42 c10::intrusive_ptr<Work> allreduce_coalesced( in allreduce_coalesced() function in c10d::FakeProcessGroup
H A DBackend.hpp109 virtual c10::intrusive_ptr<Work> allreduce_coalesced( in allreduce_coalesced() function in c10d::Backend
H A DProcessGroup.hpp176 virtual c10::intrusive_ptr<Work> allreduce_coalesced( in allreduce_coalesced() function in c10d::ProcessGroup
H A DProcessGroupWrapper.cpp413 c10::intrusive_ptr<Work> ProcessGroupWrapper::allreduce_coalesced( in allreduce_coalesced() function in c10d::ProcessGroupWrapper
H A DProcessGroupMPI.cpp442 c10::intrusive_ptr<Work> ProcessGroupMPI::allreduce_coalesced( in allreduce_coalesced() function in c10d::ProcessGroupMPI
H A DProcessGroupUCC.cpp1031 c10::intrusive_ptr<Work> ProcessGroupUCC::allreduce_coalesced( in allreduce_coalesced() function in c10d::ProcessGroupUCC
H A DProcessGroupNCCL.cpp3484 c10::intrusive_ptr<Work> ProcessGroupNCCL::allreduce_coalesced( in checkForNCCLErrorsInternal() function in c10d::ProcessGroupNCCL
H A DProcessGroupGloo.cpp1557 c10::intrusive_ptr<Work> ProcessGroupGloo::allreduce_coalesced( in allreduce_coalesced() function in c10d::ProcessGroupGloo
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Dmulti_threaded_pg.py342 def allreduce_coalesced(self, tensor_list, opts=AllreduceOptions()): member in ProcessLocalGroup