pytorch
00b0d826 - [ONNX] shape type inference fixes for control flow (#59319) (#60248)

Commit
3 years ago
[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>
Author
Parents
Loading