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.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We鈥檒l occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Support torch.nn.functional.adaptive_avg_pool2d #346

Closed
kiya00 opened this issue May 2, 2024 · 1 comment 路 Fixed by #363
Closed

Support torch.nn.functional.adaptive_avg_pool2d #346

kiya00 opened this issue May 2, 2024 · 1 comment 路 Fixed by #363
Assignees
@kiya00
Labels
enhancement New feature or request operators

Comments

@kiya00
Copy link
Collaborator

kiya00 commented May 2, 2024 •
edited by github-actions bot

馃殌 Feature

Support torch.nn.functional.adaptive_avg_pool2d that is used in ResNet50

Motivation

Support ResNet50

cc @apaz-cli
@kiya00 kiya00 added enhancement New feature or request operators labels May 2, 2024
@nikitaved
Copy link
Contributor

This will require a prim for sure!

@kiya00 kiya00 self-assigned this May 2, 2024
kiya00 added a commit that referenced this issue May 3, 2024
@kiya00
Add torch.nn.functional.adaptive_avg_pool2d (#346)
be26e16
@kiya00 kiya00 mentioned this issue May 3, 2024
Merged
4 tasks
@t-vi t-vi closed this as completed in #363 May 21, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@kiya00 kiya00

Labels
enhancement New feature or request operators
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

add adaptive_avg_pool2d Lightning-AI/lightning-thunder
2 participants
@nikitaved @kiya00