Searched defs:dist_init (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/ | ||
| H A D | test_functional_api.py | 482 def dist_init(self): member in TestCollectivesWithNCCL |
| H A D | test_c10d_logger.py | 80 def dist_init(self): member in C10dErrorLoggerTest |
| H A D | test_c10d_object_collectives.py | 75 def dist_init(self): member in TestObjectCollectives |
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ | ||
| H A D | dist_utils.py | 22 def dist_init( function |
| /aosp_15_r20/external/pytorch/test/distributed/algorithms/ | ||
| H A D | test_join.py | 172 def dist_init(self, rank, world_size, backend=BACKEND): member in TestJoin |
| /aosp_15_r20/external/pytorch/test/distributed/optim/ | ||
| H A D | test_zero_redundancy_optimizer.py | 94 def dist_init(self, rank, world_size=-1, backend=BACKEND): member in TestZeroRedundancyOptimizer |