Searched defs:optim_cls (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/optim/ |
| H A 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
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_compiled_optimizers.py | 318 optim_cls, argument
352 optim_cls, argument
442 def make_recompile_test(optim_cls, closure=None, kernel_count=2, **kwargs): argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/ |
| H A D | dist_optimizer_test.py | 142 def _test_dist_optim_base(self, optim_cls, *args, **kwargs): argument
215 def _test_dist_optim_none_grads(self, optim_cls, *args, **kwargs): argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/algorithms/_optimizer_overlap/ |
| H A D | optimizer_overlap.py | 21 def register_overlapped(optim_cls): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_functional_optim.py | 91 def _test_functional_optim_parity(self, optim_cls, *args, **kwargs): argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | distributed_test.py | 4555 optim_cls, argument
4791 optim_cls, argument
|