[ONNX] shape type inference fixes for control flow (#59319) (#60248)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/60248
* ~~Allow shape inference to skip for blocks by checking unsupported cases recursively. Currently onnx::Identity would trigger a shape inference failure.~~ Fixed in onnx submodule 1.9.
* Remove previous special post process for if op, since that was for constant folding, and now it is handled elsewhere. Update new post process for control flow nodes to copy assign node shape from subblock output shape correctly.
Test Plan: Imported from OSS
Reviewed By: zou3519, ZolotukhinM
Differential Revision: D29494913
Pulled By: SplitInfinity
fbshipit-source-id: de274a388df86e86403981e1b89b8b4a0d1e26d1
Co-authored-by: BowenBao <bowbao@microsoft.com>