Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupMPI.cpp642 at::Tensor flatInputTensor; in scatter() local
H A DProcessGroupGloo.cpp1829 at::Tensor flatInputTensor = flattenDenseTensors(inputs); in allgather() local
2103 at::Tensor flatInputTensor = flattenDenseTensors(input_list); in allgather_coalesced() local