Home
last modified time | relevance | path

Searched refs:flattenDenseTensors (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/csrc/distributed/c10d/
DUtils.hpp411 inline at::Tensor flattenDenseTensors(at::TensorList tensors) { in flattenDenseTensors() function
DProcessGroupGloo.cpp1086 at::Tensor coalescedTensor = flattenDenseTensors(tensors); in allreduceCoalesced()
1829 at::Tensor flatInputTensor = flattenDenseTensors(inputs); in allgather()
2103 at::Tensor flatInputTensor = flattenDenseTensors(input_list); in allgather_coalesced()