Searched refs:AllGatherIntoTensor (Results 1 – 1 of 1) sorted by relevance
516 class AllGatherIntoTensor class517 : public torch::autograd::Function<AllGatherIntoTensor> {567 return AllGatherIntoTensor::apply(input, group_size, group_name); in all_gather_into_tensor_autograd()