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

torch.aten.max_unpool3d #765

Closed
Tracked by #347
Hardcode84 opened this issue Jul 8, 2024 · 2 comments
Closed
Tracked by #347

torch.aten.max_unpool3d #765

Hardcode84 opened this issue Jul 8, 2024 · 2 comments
Assignees

Comments

@Hardcode84
Copy link

No description provided.

@Hardcode84
Copy link
Author

llvm/torch-mlir#3536

@Hardcode84
Copy link
Author

The PR was merged but there are known limitations in current implementation, specifically, case when the original pooling had overlapping windows is not supported. aten op doesn't preserve enough info so we will need a completely different approach and/or direct ONNX->linalg lowering to handle it (see technical details in the PR).

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

No branches or pull requests

1 participant