Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/distributed/fsdp/
H A Dtest_fsdp_mixed_precision.py423 test_reduce_scatter = partial(
429 with patch_reduce_scatter(test_reduce_scatter, full_precision_param_dtype):
592 test_reduce_scatter = partial(
598 with patch_reduce_scatter(test_reduce_scatter, param_dtype):
974 test_reduce_scatter = partial(
981 with patch_reduce_scatter(test_reduce_scatter, torch.float32):
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_spawn_nccl.py154 def test_reduce_scatter(self): member in TestDistributedNNFunctionsNccl
H A Dtest_fake_pg.py54 def test_reduce_scatter(self): member in TestFakePG
H A Dtest_nccl.py207 def test_reduce_scatter(self, device, dtype): member in TestNCCL
H A Dtest_multi_threaded_pg.py176 def test_reduce_scatter(self): member in TestCollectivesWithBaseClass
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
H A Dimport.hlotxt952 // CHECK-LABEL: func private @test_reduce_scatter
954 %test_reduce_scatter {