Searched defs:flatInputTensor (Results 1 – 2 of 2) sorted by relevance
642 at::Tensor flatInputTensor; in scatter() local
1829 at::Tensor flatInputTensor = flattenDenseTensors(inputs); in allgather() local2103 at::Tensor flatInputTensor = flattenDenseTensors(input_list); in allgather_coalesced() local