Create prototype for AST rewriter (#47216)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/47216
Test Plan: Imported from OSS
Reviewed By: glaringlee
Differential Revision: D24687539
Pulled By: ansley
fbshipit-source-id: 421108d066ff93ee18f4312ee67c287ca1cef881
