onnx softmaxoutput fix#13116
Conversation
|
@mxnet-label-bot[pr-awaiting-review, ONNX] |
|
@Roshrini can you give more context why this change is needed. The link you provided just says that there is a bug. What is the bug, how was it introduced, and how is this fixing it? @zhreshold for review. |
|
get_inputs(node, kwargs) method which was getting used works on all the inputs whereas for SoftmaxOutput operator we are using single input. So, iterating on all the inputs gave OutOfIndex error. |
|
@zhreshold Can you review/merge this PR? |
|
@mxnet-label-bot update [pr-awaiting-merge] |
sandeep-krishnamurthy
left a comment
There was a problem hiding this comment.
LGTM.
Why was this not caught in tests?
Are there any tests added now?
|
@mxnet-label-bot update [pr-work-in-progress] |
|
@mxnet-label-bot add [pr-awaiting-testing] |
| and return the created node. | ||
| """ | ||
| name, _, _ = get_inputs(node, kwargs) | ||
| name = node["name"] |
There was a problem hiding this comment.
The change LGTM, can you add a test case to guard this against happening again?
|
@sandeep-krishnamurthy @zhreshold Added test case for the operator |
Description
Fix for the failing tutorial mentioned here
#13099
@vandanavk
Checklist
Essentials