pytorch
24ff92f7 - [ONNX] Redesign inplace conversion (#55033) (#56173)

Commit
3 years ago
[ONNX] Redesign inplace conversion (#55033) (#56173) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56173 * Create `InplaceConverter` and `ValueTracker` to keep track of aliases of values throughout the graph. For a given value, a new alias is created every time when there is an inplace operation, SetAttr, or through nested blocks owned by If/Loop nodes. * Fix bug where controlflow node output types are not set, when the complete node is unable to run ONNX shape inference due to containing non-onnx node. * Add symbolic for `__not__` ~~and `prim_min`~~(update: moved to a separate PR), and update `index_put` opset9 to support case of assignment without providing indices. * Bump ORT version in CI test. Test Plan: Imported from OSS Reviewed By: pbelevich Differential Revision: D27866138 Pulled By: SplitInfinity fbshipit-source-id: ab5c9188740c50f783ceba4d54fda43c26e2fde7
Author
Parents
Loading