Home
last modified time | relevance | path

Searched refs:test_all_to_all (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_spawn_ucc.py119 def test_all_to_all(self): member in TestDistributedNNFunctionsUcc
H A Dtest_c10d_spawn_nccl.py137 def test_all_to_all(self): member in TestDistributedNNFunctionsNccl
H A Dtest_c10d_spawn_gloo.py261 def test_all_to_all(self): member in TestDistributedNNFunctionsGloo
H A Dtest_multi_threaded_pg.py203 def test_all_to_all(self): member in TestCollectivesWithBaseClass
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
H A Dimport.hlotxt57 // CHECK-LABEL: func private @test_all_to_all
59 %test_all_to_all {
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Ddistributed_test.py3932 def test_all_to_all(self): member in DistributedTest._DistTestBase