[jit] do the code reorg (#33851)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/33851
Rationale and context described in #33828.
Script to reproduce the move:
https://gist.github.com/suo/16cbefaaeb67ca5a7c6caffd49b7f6e9
ghstack-source-id: 99079645
Test Plan: Make sure CI passes
Reviewed By: jamesr66a
Differential Revision: D20133869
fbshipit-source-id: 390e9241a9c85366d9005c492ac31f10aa96488e