Home
last modified time | relevance | path

Searched refs:reduce_scatter_tensor_autograd (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DFunctional.cpp508 at::Tensor reduce_scatter_tensor_autograd( in reduce_scatter_tensor_autograd() function
588 c10::DispatchKey::Autograd, ::reduce_scatter_tensor_autograd), in TORCH_LIBRARY()
/aosp_15_r20/external/pytorch/torch/distributed/
H A D_functional_collectives.py293 def reduce_scatter_tensor_autograd( function
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_functional_api.py761 ft_c.reduce_scatter_tensor_autograd(