pytorch

5c893375 - refactor autograd_hook

refactor autograd_hook put part of codes in autograd_hook into functions, so that they can be used in the static graph training later on. Differential Revision: [D27439508](https://our.internmc.facebook.com/intern/diff/D27439508/) [ghstack-poisoned]