pytorch
9d1263a8 - [ONNX] Fix Gather replacement in RNN peephole (#93120)

Commit
2 years ago
[ONNX] Fix Gather replacement in RNN peephole (#93120) From PR: https://github.com/pytorch/pytorch/pull/58691, Replacing the second input of `Gather` 0 to 1 affects other innocent Nodes. In Issue #91526 onnx::range starts from 0, the 0 is changed by this mechanism, as it's shared with onnx::Gather. This PR intends to create a whole independent Constant 0 for replacement. NOTE: The PR passes all existing RNN tests locally in case CI doesn't include RNN test. ~~TODO: test~~ Pull Request resolved: https://github.com/pytorch/pytorch/pull/93120 Approved by: https://github.com/BowenBao
Author
Committer
Parents
Loading