Searched refs:test_allreduce (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/ | ||
| H A D | test_c10d_spawn_ucc.py | 100 def test_allreduce(self): member in TestDistributedNNFunctionsUcc |
| H A D | test_c10d_spawn_nccl.py | 121 def test_allreduce(self): member in TestDistributedNNFunctionsNccl |
| H A D | test_c10d_spawn_gloo.py | 245 def test_allreduce(self): member in TestDistributedNNFunctionsGloo |