Home
last modified time | relevance | path

Searched refs:_test_all_to_all_single (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/algorithms/quantization/
H A Dtest_quantization.py174 self._test_all_to_all_single(
198 self._test_all_to_all_single(
275 def _test_all_to_all_single( member in DistQuantizationTests
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_spawn_ucc.py128 self._test_all_to_all_single("ucc")
H A Dtest_c10d_spawn_nccl.py146 self._test_all_to_all_single("nccl")
H A Dtest_c10d_spawn.py228 def _test_all_to_all_single(self, backend): member in TestDistributedNNFunctions
H A Dtest_c10d_spawn_gloo.py270 self._test_all_to_all_single("gloo")
H A Dtest_c10d_common.py1912 def _test_all_to_all_single(self, backend): member in ProcessGroupWithDispatchedCollectivesTests
H A Dtest_c10d_gloo.py2438 self._test_all_to_all_single(backend="gloo")
H A Dtest_c10d_nccl.py3130 self._test_all_to_all_single(backend="nccl")