Home
last modified time | relevance | path

Searched defs:recv_offsets (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/csrc/distributed/c10d/
DProcessGroupUCC.hpp79 std::vector<uint64_t> recv_offsets; member in c10d::ProcessGroupUCC::AlltoallWorkData
86 std::vector<uint64_t> recv_offsets; member in c10d::ProcessGroupUCC::AllgathervWorkData
DProcessGroupMPI.cpp750 std::vector<int> recv_offsets(size_); in alltoall_base() local
794 std::vector<int> recv_offsets(size_); in alltoall() local
DProcessGroupNCCL.cpp4229 std::vector<size_t> recv_offsets(size_); in checkForNCCLErrorsInternal() local