pytorch/torch/distributed/optim
2024-07-16 01:24:38 +00:00
..
__init__.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
apply_optimizer_in_backward.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
functional_adadelta.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
functional_adagrad.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
functional_adam.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
functional_adamax.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
functional_adamw.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
functional_rmsprop.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
functional_rprop.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
functional_sgd.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
named_optimizer.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
optimizer.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
post_localSGD_optimizer.py Flip default value for mypy disallow_untyped_defs [6/11] (#127843) 2024-06-08 18:49:29 +00:00
utils.py [BE][Easy] enable UFMT for torch/distributed/{fsdp,optim,rpc}/ (#128869) 2024-06-18 21:49:08 +00:00
zero_redundancy_optimizer.py Pass torch.load(weights_only=) internally to avoid FutureWarning (#130663) 2024-07-16 01:24:38 +00:00
zero_redundancy_optimizer.pyi Use Generic TypeAlias (PEP 585) and Union Type (PEP 604) in .pyi stub files (#129419) 2024-06-29 09:23:39 +00:00