-
Notifications
You must be signed in to change notification settings - Fork 25.7k
Replace use of ConstantLike with with ConstantOfShape #16095
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Replace use of ConstantLike with with ConstantOfShape #16095
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Overall looks good, but I need to add Caffe2 ConstantOfShape op to Caffe2. Otherwise, the end to end test on RNN models won't work.
|
@houseroad - There's some failure in test run and I am not sure I understand why. Could you please take a look and advise. |
|
Yes, I am working on that |
c28ac61 to
1bfa277
Compare
1bfa277 to
3d0f6e3
Compare
3d0f6e3 to
d18bf90
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good. Thanks.
Summary: Submitting this PR as an update to existing PR (pytorch/pytorch#15938) on houseroad 's request. This PR replaces the use of ONNX op `ConstantLike` with `ConstantOfShape` in the ONNX exporter. In addition to removing the call sites in `symbolic.py`, it also replace the call site in `peephole.cpp`. Pull Request resolved: pytorch/pytorch#16095 Differential Revision: D13745723 Pulled By: houseroad fbshipit-source-id: e2a5f534f01adf199df9e27544f7afcfa540e1f0
Submitting this PR as an update to existing PR (#15938) on @houseroad 's request.
This PR replaces the use of ONNX op
ConstantLikewithConstantOfShapein the ONNX exporter. In addition to removing the call sites insymbolic.py, it also replace the call site inpeephole.cpp.