Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupMPI.cpp503 auto flatOutputTensor = newLikeFlat(outputDataVec); in allgather() local
565 at::Tensor flatOutputTensor; in gather() local
H A DProcessGroupGloo.cpp1835 at::Tensor flatOutputTensor = newLikeFlat(outputs[0]); in allgather() local
2114 at::Tensor flatOutputTensor = in allgather_coalesced() local
2248 at::Tensor flatOutputTensor; in gather() local