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_unpool2d #764

Open
Tracked by #347
Hardcode84 opened this issue Jul 3, 2024 · 3 comments
Open
Tracked by #347

torch.aten.max_unpool2d #764

Hardcode84 opened this issue Jul 3, 2024 · 3 comments

Comments

@Hardcode84
Copy link

No description provided.

@Hardcode84
Copy link
Author

Hardcode84 commented Jul 30, 2024

PR for max_unpool3d was merged (#765), but unfortunately, this approach won't work for max_unpool2d as corresponding aten node preserve even less info than max_unpool3d, we need a different approach, either lowering directly to (parallel) scf loops or ONNX->linalg lowering, skipping aten op. See max_unpool3d issue and PR for details.

@Hardcode84 Hardcode84 removed their assignment Jul 30, 2024
@vivekkhandelwal1
Copy link
Contributor

Hi @Hardcode84, are you still working on this op?

@Hardcode84
Copy link
Author

Hi, @vivekkhandelwal1, I did max_unpool3d but I'm not working on max_unpool2d currently.

@marbre marbre assigned marbre and unassigned marbre Aug 16, 2024
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

3 participants