pytorch
1c9347c6 - [ONNX] Use parameter values in onnx shape inference (#49706) (#50905)

Commit
3 years ago
[ONNX] Use parameter values in onnx shape inference (#49706) (#50905) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/50905 Adds an additional run of onnx shape inference after constant folding, since initializer may have changed and affected shape inference. Test Plan: Imported from OSS Reviewed By: pbelevich Differential Revision: D26050881 Pulled By: SplitInfinity fbshipit-source-id: 9e5d69c52b647133cd3a0781988e2ad1d1a9c09d
Author
Parents
Loading