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’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

add channels last for AdaptiveAvgPool2d #48916

Closed
wants to merge 1 commit into from
Closed

add channels last for AdaptiveAvgPool2d #48916

wants to merge 1 commit into from

Commits on Dec 7, 2020

  1. add channels last for AdaptiveAvgPool2d 

    optimize adaptive average pool2d forward path

optimize adaptive average pool2d backward path

remove unused headers

minor change

minor change

rename the header; add adaptive max pooling in future.

minor change

loosen adapative_pool2d test on nhwc to both device cuda and cpu

minor change

[ghstack-poisoned]
    @mingfeima
    mingfeima committed Dec 7, 2020
    Configuration menu
    Copy the full SHA
    06a3207 View commit details
    Browse the repository at this point in the history