Move unboxing for factory ops to after dispatch (#36564)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/36564
-
ghstack-source-id: 102292162
Test Plan: waitforsandcastle
Differential Revision: D21014324
fbshipit-source-id: 5b95eafbb668bed174cd2c826e308e74e329f552