Home
last modified time | relevance | path

Searched defs:reduce_scatter_tensor (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/_tools/
H A Dfsdp2_mem_tracker.py494 def reduce_scatter_tensor( function
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DFunctional.cpp247 at::Tensor reduce_scatter_tensor( in reduce_scatter_tensor() function
/aosp_15_r20/external/pytorch/torch/distributed/
H A D_functional_collectives.py251 def reduce_scatter_tensor( function
H A Ddistributed_c10d.py3766 def reduce_scatter_tensor(output, input, op=ReduceOp.SUM, group=None, async_op=False): function