pytorch
51b7224f - [vulkan] Add max_pool2d op (#58806)

Commit
3 years ago
[vulkan] Add max_pool2d op (#58806) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58806 Adds the max_pool2d op to Vulkan. Test Plan: ``` cd ~/fbsource buck build -c ndk.custom_libcxx=false -c pt.enable_qpl=0 //xplat/caffe2:pt_vulkan_api_test_binAndroid\#android-arm64 --show-output adb push buck-out/gen/xplat/caffe2/pt_vulkan_api_test_binAndroid\#android-arm64 /data/local/tmp/vulkan_api_test adb shell "/data/local/tmp/vulkan_api_test" cd - ``` Reviewed By: IvanKobzarev Differential Revision: D28625049 fbshipit-source-id: 75c82a84f0eca51627e33a6182ef51cb7e82e068
Author
Parents
Loading