[ONNX] Support nonzero(*, as_tuple=True) export #47421
Conversation
facebook-github-bot
added
cla signed
oncall: jit
💊 CI failures summary and remediationsAs of commit 5755653 (more details on the Dr. CI page):
ci.pytorch.org: 1 failedThis comment was automatically generated by Dr. CI (expand for details).Follow this link to opt-out of these comments for your Pull Requests.Please report bugs/suggestions on the GitHub issue tracker or post in the (internal) Dr. CI Users group. This comment has been revised 14 times. |
heitorschueroff
added
triaged
There was a problem hiding this comment.
@bzinodev has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
Test failure seems related, I will resolve the test and ping you when this is ready. @bzinodev |
There was a problem hiding this comment.
@bzinodev has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
Codecov Report
@@ Coverage Diff @@
## master #47421 +/- ##
=======================================
Coverage 64.59% 64.59%
=======================================
Files 1680 1680
Lines 166920 166929 +9
=======================================
+ Hits 107822 107833 +11
+ Misses 59098 59096 -2 |
There was a problem hiding this comment.
@bzinodev 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.
@bzinodev 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.
@bzinodev has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
Support exporting with
as_tuple = trueExample:
torch.nonzero(x, as_tuple=True)This is the same as
torch.unbind(torch.nonzero(x), 1)