pytorch

ae52750d - Reduce hook registration code duplication (#91418)

Reduce hook registration code duplication (#91418) Pull Request resolved: https://github.com/pytorch/pytorch/pull/91418 Approved by: https://github.com/albanD