Searched defs:_test_all_to_all_single (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/algorithms/quantization/ | ||
| H A D | test_quantization.py | 275 def _test_all_to_all_single( member in DistQuantizationTests |
| /aosp_15_r20/external/pytorch/test/distributed/ | ||
| H A D | test_c10d_spawn.py | 228 def _test_all_to_all_single(self, backend): member in TestDistributedNNFunctions |
| H A D | test_c10d_common.py | 1912 def _test_all_to_all_single(self, backend): member in ProcessGroupWithDispatchedCollectivesTests |