Home
last modified time | relevance | path

Searched defs:_reduce_scatter_base (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DFakeProcessGroup.hpp128 c10::intrusive_ptr<Work> _reduce_scatter_base( in _reduce_scatter_base() function in c10d::FakeProcessGroup
H A DBackend.hpp211 virtual c10::intrusive_ptr<Work> _reduce_scatter_base( in _reduce_scatter_base() function in c10d::Backend
H A DProcessGroup.hpp371 virtual c10::intrusive_ptr<Work> _reduce_scatter_base( in _reduce_scatter_base() function in c10d::ProcessGroup
H A DProcessGroupWrapper.cpp557 c10::intrusive_ptr<Work> ProcessGroupWrapper::_reduce_scatter_base( in _reduce_scatter_base() function in c10d::ProcessGroupWrapper
H A DProcessGroupNCCL.cpp3964 c10::intrusive_ptr<Work> ProcessGroupNCCL::_reduce_scatter_base( in checkForNCCLErrorsInternal() function in c10d::ProcessGroupNCCL
H A DProcessGroupGloo.cpp1943 c10::intrusive_ptr<Work> ProcessGroupGloo::_reduce_scatter_base( in _reduce_scatter_base() function in c10d::ProcessGroupGloo
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Dmulti_threaded_pg.py385 def _reduce_scatter_base(self, output_tensor, input_tensor, opts=ReduceScatterOptions()): member in ProcessLocalGroup
/aosp_15_r20/external/pytorch/torch/distributed/
H A Ddistributed_c10d.py3856 def _reduce_scatter_base(output, input, op=ReduceOp.SUM, group=None, async_op=False): function