Home
last modified time | relevance | path

Searched refs:allreduce_with_then_hook (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_nccl.py2019 def allreduce_with_then_hook( function
2037 process_group, allreduce_with_then_hook
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Ddistributed_test.py5409 def allreduce_with_then_hook( function
5425 num_iters=4, ddp_comm_hook=allreduce_with_then_hook