Searched defs:join_hook (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/torch/distributed/algorithms/ | ||
| D | join.py | 61 def join_hook(self, **kwargs) -> JoinHook: member in Joinable |
| /external/pytorch/test/distributed/algorithms/ | ||
| D | test_join.py | 109 def join_hook(self, **kwargs) -> JoinHook: member in AllReducer |
| /external/pytorch/torch/distributed/optim/ | ||
| D | zero_redundancy_optimizer.py | 1120 def join_hook(self, **kwargs): member in ZeroRedundancyOptimizer |
| /external/pytorch/test/distributed/optim/ | ||
| D | test_zero_redundancy_optimizer.py | 1072 def join_hook(self, **kwargs): member in TestZeroRedundancyOptimizerDistributed._test_zero_join._GradientSetter |
| /external/pytorch/torch/nn/parallel/ | ||
| D | distributed.py | 1848 def join_hook( member in DistributedDataParallel |