Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/distributed/c10d/
DProcessGroupUCC.hpp76 std::vector<uint64_t> send_lengths; member in c10d::ProcessGroupUCC::AlltoallWorkData
92 std::vector<uint64_t> send_lengths; member in c10d::ProcessGroupUCC::ScattervWorkData
DProcessGroupMPI.cpp747 std::vector<int> send_lengths(size_); in alltoall_base() local
791 std::vector<int> send_lengths(size_); in alltoall() local
DProcessGroupNCCL.cpp4226 std::vector<size_t> send_lengths(size_); in checkForNCCLErrorsInternal() local