Move some class/functions in test_jit.py to jit_utils.py (#26839)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/26839
att
Test Plan:
ci
Imported from OSS
Differential Revision: D17643010
fbshipit-source-id: 5768b70410b7bdfdbee734d3a00296e5b1ad30d5