pytorch
dc0071df - [ONNX] Special post process for onnx::Cast and onnx::ConstantOfShape shape type inference (#55962) (#57597)

Commit
4 years ago
[ONNX] Special post process for onnx::Cast and onnx::ConstantOfShape shape type inference (#55962) (#57597) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/57597 * Special post process for onnx::Cast and onnx::ConstantOfShape * Update `test_pytorch_onnx_shape_inference.py` to be unit test over shape inference patterns. Test Plan: Imported from OSS Reviewed By: malfet Differential Revision: D28393529 Pulled By: SplitInfinity fbshipit-source-id: fc26032ddb842d4e299447da39564b28049752ed Co-authored-by: BowenBao <bowbao@microsoft.com>
Author
Parents
Loading