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

Add onnx op LRN lowering #3432

Merged
merged 1 commit into from
Jun 14, 2024
Merged

Conversation

manupak
Copy link
Contributor

@manupak manupak commented Jun 7, 2024

This commit adds support for lowering
Onnx LRN op to aten.

@manupak
Copy link
Contributor Author

manupak commented Jun 7, 2024

e2e testing : nod-ai/SHARK-TestSuite#255

@manupak
Copy link
Contributor Author

manupak commented Jun 7, 2024

cc: @vivekkhandelwal1

lib/Dialect/Torch/Utils/Utils.cpp Outdated Show resolved Hide resolved
include/torch-mlir/Dialect/Torch/Utils/Utils.h Outdated Show resolved Hide resolved
lib/Conversion/TorchOnnxToTorch/DefaultDomainGtoP.cpp Outdated Show resolved Hide resolved
lib/Conversion/TorchOnnxToTorch/DefaultDomainGtoP.cpp Outdated Show resolved Hide resolved
lib/Conversion/TorchOnnxToTorch/DefaultDomainGtoP.cpp Outdated Show resolved Hide resolved
lib/Conversion/TorchOnnxToTorch/DefaultDomainGtoP.cpp Outdated Show resolved Hide resolved
lib/Conversion/TorchOnnxToTorch/DefaultDomainGtoP.cpp Outdated Show resolved Hide resolved
This commit adds support for lowering
Onnx LRN op to aten.
Copy link
Collaborator

@vivekkhandelwal1 vivekkhandelwal1 left a comment

Choose a reason for hiding this comment

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

LGTM

@vivekkhandelwal1 vivekkhandelwal1 enabled auto-merge (squash) June 14, 2024 16:29
@vivekkhandelwal1 vivekkhandelwal1 merged commit d2b663e into llvm:main Jun 14, 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.

2 participants