Home
last modified time | relevance | path

Searched refs:allreduce_sparse (Results 1 – 8 of 8) 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 DProcessGroupGloo.hpp292 c10::intrusive_ptr<Work> allreduce_sparse(
H A DOps.cpp467 ->allreduce_sparse( in allreduce_sparse_cuda_()
H A DProcessGroupNCCL.hpp553 c10::intrusive_ptr<Work> allreduce_sparse(
H A DProcessGroupGloo.cpp1548 c10::intrusive_ptr<Work> ProcessGroupGloo::allreduce_sparse( in allreduce_sparse() function in c10d::ProcessGroupGloo
H A DProcessGroupNCCL.cpp3337 c10::intrusive_ptr<Work> ProcessGroupNCCL::allreduce_sparse( in checkForNCCLErrorsInternal() function in c10d::ProcessGroupNCCL
/aosp_15_r20/external/pytorch/test/cpp/c10d/
H A DProcessGroupNCCLTest.cpp260 auto results = pg_->allreduce_sparse(tensors_); in run()