add OpInfo for torch.nn.functional.one_hot#62253
Conversation
🔗 Helpful links
💊 CI failures summary and remediationsAs of commit 06173d0 (more details on the Dr. CI page):
🕵️ 2 new failures recognized by patternsThe following CI failures do not appear to be due to upstream breakages:
|
krshrimali
left a comment
There was a problem hiding this comment.
LGTM, @pmeier! Thanks. Left a minor suggestion, but it's not a necessary change. :)
zou3519
left a comment
There was a problem hiding this comment.
This looks pretty good. I added some suggestions for some more test cases to make the OpInfo more complete
|
The lint failure seems bogus so I am going to land this: |
|
@zou3519 has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
Addresses pytorch/functorch#78.