pytorch

99244435 - Resolve TODO after Python 2 for custom_fwd (#78592)

Resolve TODO after Python 2 for custom_fwd (#78592) Pull Request resolved: https://github.com/pytorch/pytorch/pull/78592 Approved by: https://github.com/seemethere