pytorch/torch/optim/_multi_tensor
Iurii Zdebskyi e7564b076c Refactor scalar list APIs to use overloads (#45673)
Summary:
Refactor foreach APIs to use overloads in case of scalar list inputs.
Tested via unit tests.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/45673

Reviewed By: heitorschueroff

Differential Revision: D24053424

Pulled By: izdeby

fbshipit-source-id: 35976cc50b4acfe228a32ed26cede579d5621cde
2020-10-19 09:28:49 -07:00
..
__init__.py [RELAND] Added optimizers based on multi tensor apply (#45408) 2020-09-28 13:14:04 -07:00
__init__.pyi [RELAND] Added optimizers based on multi tensor apply (#45408) 2020-09-28 13:14:04 -07:00
adadelta.py [RELAND] Added optimizers based on multi tensor apply (#45408) 2020-09-28 13:14:04 -07:00
adadelta.pyi [RELAND] Added optimizers based on multi tensor apply (#45408) 2020-09-28 13:14:04 -07:00
adam.py Refactor scalar list APIs to use overloads (#45673) 2020-10-19 09:28:49 -07:00
adam.pyi [RELAND] Added optimizers based on multi tensor apply (#45408) 2020-09-28 13:14:04 -07:00
adamax.py Added scalar lists APIs for addcdiv and addcmul (#45932) 2020-10-14 08:12:37 -07:00
adamax.pyi [RELAND] Added optimizers based on multi tensor apply (#45408) 2020-09-28 13:14:04 -07:00
adamw.py Refactor scalar list APIs to use overloads (#45673) 2020-10-19 09:28:49 -07:00
adamw.pyi [RELAND] Added optimizers based on multi tensor apply (#45408) 2020-09-28 13:14:04 -07:00
asgd.py [RELAND] Added optimizers based on multi tensor apply (#45408) 2020-09-28 13:14:04 -07:00
asgd.pyi [RELAND] Added optimizers based on multi tensor apply (#45408) 2020-09-28 13:14:04 -07:00
rmsprop.py [RELAND] Added optimizers based on multi tensor apply (#45408) 2020-09-28 13:14:04 -07:00
rmsprop.pyi [RELAND] Added optimizers based on multi tensor apply (#45408) 2020-09-28 13:14:04 -07:00
rprop.py [RELAND] Added optimizers based on multi tensor apply (#45408) 2020-09-28 13:14:04 -07:00
rprop.pyi [RELAND] Added optimizers based on multi tensor apply (#45408) 2020-09-28 13:14:04 -07:00
sgd.py [RELAND] Added optimizers based on multi tensor apply (#45408) 2020-09-28 13:14:04 -07:00
sgd.pyi [RELAND] Added optimizers based on multi tensor apply (#45408) 2020-09-28 13:14:04 -07:00