Register Saved Tensors hooks (#60663)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/60663
Test Plan: Imported from OSS
Reviewed By: soulitzer
Differential Revision: D29466223
fbshipit-source-id: 65dc3a935c18a0e6b93a37e24543c696e6ae0321
