Home
last modified time | relevance | path

Searched defs:allgather_into_tensor_coalesced (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DPyProcessGroup.hpp82 c10::intrusive_ptr<Work> allgather_into_tensor_coalesced( in allgather_into_tensor_coalesced() function in c10d::PyProcessGroup
H A DFakeProcessGroup.hpp93 c10::intrusive_ptr<Work> allgather_into_tensor_coalesced( in allgather_into_tensor_coalesced() function in c10d::FakeProcessGroup
H A DBackend.hpp171 virtual c10::intrusive_ptr<Work> allgather_into_tensor_coalesced( in allgather_into_tensor_coalesced() function in c10d::Backend
H A DProcessGroup.hpp287 virtual c10::intrusive_ptr<Work> allgather_into_tensor_coalesced( in allgather_into_tensor_coalesced() function in c10d::ProcessGroup
H A DProcessGroupNCCL.cpp3834 c10::intrusive_ptr<Work> ProcessGroupNCCL::allgather_into_tensor_coalesced( in checkForNCCLErrorsInternal() function in c10d::ProcessGroupNCCL
H A DProcessGroupGloo.cpp2194 c10::intrusive_ptr<Work> ProcessGroupGloo::allgather_into_tensor_coalesced( in allgather_into_tensor_coalesced() function in c10d::ProcessGroupGloo
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Dmulti_threaded_pg.py399 …def allgather_into_tensor_coalesced(self, output_tensor_list, input_tensor_list, opts=AllgatherOpt… member in ProcessLocalGroup