Searched defs:local_params_rref (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/distributed/optim/ |
| D | optimizer.py | 46 def __init__(self, optim_cls, local_params_rref, *args, **kwargs): argument
75 def __init__(self, optim_cls, local_params_rref, *args, **kwargs): argument
88 def _new_local_optimizer(optim_cls, local_params_rref, *args, **kwargs): argument
98 def _new_script_local_optimizer(optim_cls, local_params_rref, *args, **kwargs): argument
|