Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/distributed/c10d/
DProcessGroupUCC.hpp78 std::vector<uint64_t> recv_lengths; member in c10d::ProcessGroupUCC::AlltoallWorkData
85 std::vector<uint64_t> recv_lengths; member in c10d::ProcessGroupUCC::AllgathervWorkData
DProcessGroupMPI.cpp748 std::vector<int> recv_lengths(size_); in alltoall_base() local
792 std::vector<int> recv_lengths(size_); in alltoall() local
DProcessGroupNCCL.cpp4227 std::vector<size_t> recv_lengths(size_); in checkForNCCLErrorsInternal() local