Annotate torch.nn.cpp (#46490)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/46489
Pull Request resolved: https://github.com/pytorch/pytorch/pull/46490
Reviewed By: zhangguanheng66
Differential Revision: D24509519
Pulled By: ezyang
fbshipit-source-id: edffd32ab2ac17ae4bbd44826b71f5cb9f1da1c5
