[ONNX] Improve lower tuples and handle control flow (#57650) (#58694)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58694
Improving the logic for finding tuple patterns within control flow.
Also fixes: https://github.com/pytorch/pytorch/issues/56914
Test Plan: Imported from OSS
Reviewed By: driazati
Differential Revision: D28714806
Pulled By: SplitInfinity
fbshipit-source-id: 1552100cf9cc88e6f58df2e90758e8898ba0a9b3
Co-authored-by: neginraoof <neginmr@utexas.edu>
