Searched refs:test_all_to_all (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/ | ||
| H A D | test_c10d_spawn_ucc.py | 119 def test_all_to_all(self): member in TestDistributedNNFunctionsUcc |
| H A D | test_c10d_spawn_nccl.py | 137 def test_all_to_all(self): member in TestDistributedNNFunctionsNccl |
| H A D | test_c10d_spawn_gloo.py | 261 def test_all_to_all(self): member in TestDistributedNNFunctionsGloo |
| H A D | test_multi_threaded_pg.py | 203 def test_all_to_all(self): member in TestCollectivesWithBaseClass |
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ | ||
| H A D | import.hlotxt | 57 // CHECK-LABEL: func private @test_all_to_all |
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ | ||
| H A D | distributed_test.py | 3932 def test_all_to_all(self): member in DistributedTest._DistTestBase |