Replace AT_ASSERT with TORCH_INTERNAL_ASSERT/TORCH_CHECK (#20668)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/20668
AT_ASSERT is deprecated. Replace it wil the new exception handling macro.
Differential Revision: D15404401
fbshipit-source-id: 7ae9f5471f1f8ad95e7bb835380e098c00313d9c