Enable torch.cuda.memory typechecking (#43444)
Summary:
Add number of function prototypes defined in torch/csrs/cuda/Module.cpp to `__init__.pyi.in`
Fixes https://github.com/pytorch/pytorch/issues/43442
Pull Request resolved: https://github.com/pytorch/pytorch/pull/43444
Reviewed By: ezyang
Differential Revision: D23280221
Pulled By: malfet
fbshipit-source-id: 7d67dff7b24c8d7b7e72c919e6e7b847f242ef83