[Pytorch Edge] Prepack folding for functions besides forward (#56081)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56081
ghstack-source-id: 127205799
Test Plan: unit test. Since I'm prepacking the weights of the same operators multiple times I wonder if its a just works thing?
Reviewed By: kimishpatel
Differential Revision: D27777337
fbshipit-source-id: 909d2a667d9eb51e205536b478a6668c33b3fb15