Skip to content
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

[MLIR][ONNX] Add support for onnx.ScatterND #3479

Merged
merged 1 commit into from
Jul 8, 2024

Conversation

Shukla-Gaurav
Copy link
Collaborator

This commit adds support for onnx.ScatterND op in the onnx pipeline.

Copy link
Collaborator

@zjgarvey zjgarvey left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I only have some small questions, otherwise this looks great. Thanks for the detailed comments, since it made this much easier to review.

lib/Conversion/TorchOnnxToTorch/DefaultDomainQtoZ.cpp Outdated Show resolved Hide resolved
lib/Conversion/TorchOnnxToTorch/DefaultDomainQtoZ.cpp Outdated Show resolved Hide resolved
@Shukla-Gaurav Shukla-Gaurav force-pushed the gaurav/scatter_nd branch 2 times, most recently from 1fb2228 to 32322ec Compare July 1, 2024 16:26
This commit adds support for onnx.ScatterND op in the onnx pipeline.

Signed-Off-by: Gaurav Shukla <gaurav.shukla@amd.com>
@Shukla-Gaurav Shukla-Gaurav merged commit 0b46d11 into llvm:main Jul 8, 2024
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants