[FX] Finished prepare_for_inference API for release (#58293)
Summary:
Added an ability to configure which passes to run.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58293
Reviewed By: bdhirsh
Differential Revision: D28435948
Pulled By: Chillee
fbshipit-source-id: dfc7f1ef6b38e6f49c2423a5efe8477a645171d0