Searched refs:allocate_all_gather_output (Results 1 – 1 of 1) sorted by relevance
158 at::Tensor allocate_all_gather_output( in allocate_all_gather_output() function176 outputs.push_back(allocate_all_gather_output(tensor, group_size)); in all_gather_into_tensor_coalesced()