OpInfo JIT op.output_func handling support (#50775)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/50775
Test Plan: Imported from OSS
Reviewed By: mruberry
Differential Revision: D25964541
Pulled By: Lilyjjo
fbshipit-source-id: 8cf1ee9191d526cc46ae283f38c2d64bd60afdb2
