-
Notifications
You must be signed in to change notification settings - Fork 3.6k
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
Support ScatterND operator in ONNX #2220
Conversation
21b531d
to
d3f2798
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.
Overall, Looks good. @BowenBao, Could you please address the comment related to TF reference?
d3f2798
to
af16433
Compare
A question: I believe indices should not have a repeated value, is that correct? That is, we cannot have two or more updates for the same index-location. If so, let us add that to the documentation. (I guess this is common with scatter too, I presume.) |
af16433
to
b80ff34
Compare
@gramalingam Thanks for your detailed suggestions! I have updated the doc with clarification. You are right duplicate entries of indices is not supported. |
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.
LGTM, thanks!
@gramalingam - thanks for the review! |
@bddppq - for his review. |
@prasanthpul could you include this in ONNX 1.6 milestone as well? Thanks. |
be2d6e1
to
0548023
Compare
0548023
to
a5614a9
Compare
Support ScatterND operator in ONNX
This is the inverse of GatherND #2106
This will support the exporting of
torch.masked_scatter
,torch.index_put
,torch.copy_
,tf.scatter_nd_update
, etc.