Searched defs:send_offsets (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/csrc/distributed/c10d/ | ||
| D | ProcessGroupUCC.hpp | 77 std::vector<uint64_t> send_offsets; member in c10d::ProcessGroupUCC::AlltoallWorkData |
| D | ProcessGroupMPI.cpp | 749 std::vector<int> send_offsets(size_); in alltoall_base() local |
| D | ProcessGroupNCCL.cpp | 4228 std::vector<size_t> send_offsets(size_); in checkForNCCLErrorsInternal() local |