annotate a few torch.nn.modules.* modules (#45772)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/45771
Pull Request resolved: https://github.com/pytorch/pytorch/pull/45772
Reviewed By: mruberry
Differential Revision: D24682013
Pulled By: albanD
fbshipit-source-id: e32bc4fe9c586c079f7070924a874c70f3d127fa
