Home
last modified time | relevance | path

Searched defs:reduce_scatter_tensor_coalesced (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DPyProcessGroup.hpp168 c10::intrusive_ptr<Work> reduce_scatter_tensor_coalesced( in reduce_scatter_tensor_coalesced() function in c10d::PyProcessGroup
H A DFakeProcessGroup.hpp136 c10::intrusive_ptr<Work> reduce_scatter_tensor_coalesced( in reduce_scatter_tensor_coalesced() function in c10d::FakeProcessGroup
H A DBackend.hpp226 virtual c10::intrusive_ptr<Work> reduce_scatter_tensor_coalesced( in reduce_scatter_tensor_coalesced() function in c10d::Backend
H A DProcessGroup.hpp397 virtual c10::intrusive_ptr<Work> reduce_scatter_tensor_coalesced( in reduce_scatter_tensor_coalesced() function in c10d::ProcessGroup
H A DFunctional.cpp224 std::vector<at::Tensor> reduce_scatter_tensor_coalesced( in reduce_scatter_tensor_coalesced() function
H A DProcessGroupNCCL.cpp4040 c10::intrusive_ptr<Work> ProcessGroupNCCL::reduce_scatter_tensor_coalesced( in checkForNCCLErrorsInternal() function in c10d::ProcessGroupNCCL
H A DProcessGroupGloo.cpp1952 c10::intrusive_ptr<Work> ProcessGroupGloo::reduce_scatter_tensor_coalesced( in reduce_scatter_tensor_coalesced() function in c10d::ProcessGroupGloo
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Dmulti_threaded_pg.py389 …def reduce_scatter_tensor_coalesced(self, output_tensors, input_tensors, opts=ReduceScatterOptions… member in ProcessLocalGroup
/aosp_15_r20/external/pytorch/torch/distributed/
H A D_functional_collectives.py389 def reduce_scatter_tensor_coalesced( function