Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DUCCTracing.cpp75 const std::vector<int64_t>& outputSplitSizes, in recordOptionalInfo()
H A DPyProcessGroup.hpp121 std::vector<int64_t>& outputSplitSizes, in alltoall_base()
H A DProcessGroup.hpp422 std::vector<int64_t>& outputSplitSizes, in alltoall_base()
H A DProcessGroupWrapper.cpp495 std::vector<int64_t>& outputSplitSizes, in alltoall_base()
H A DProcessGroupMPI.cpp699 std::vector<int64_t>& outputSplitSizes, in alltoall_base()
H A DProcessGroupUCC.cpp1105 std::vector<int64_t>& outputSplitSizes, in alltoall_base()
H A Dinit.cpp2472 std::vector<int64_t> inputSplitSizes) { in c10d_init()
H A DProcessGroupNCCL.cpp4151 std::vector<int64_t>& outputSplitSizes, in checkForNCCLErrorsInternal()