Searched refs:reduce_scatter_tensor_autograd (Results 1 – 3 of 3) sorted by relevance
508 at::Tensor reduce_scatter_tensor_autograd( in reduce_scatter_tensor_autograd() function588 c10::DispatchKey::Autograd, ::reduce_scatter_tensor_autograd), in TORCH_LIBRARY()
293 def reduce_scatter_tensor_autograd( function
761 ft_c.reduce_scatter_tensor_autograd(