Home
last modified time | relevance | path

Searched defs:outputTensorLists (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroup.hpp267 std::vector<std::vector<at::Tensor>>& outputTensorLists, in allgather_coalesced()
H A DProcessGroupWrapper.cpp453 std::vector<std::vector<at::Tensor>>& outputTensorLists, in allgather_coalesced()