Implement leaky relu op
authorVincent Phan <vincentphan@fb.com>
Fri, 27 Aug 2021 20:51:38 +0000 (13:51 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Fri, 27 Aug 2021 20:52:49 +0000 (13:52 -0700)
commit358c46f99eea23ec86e4358a5d4253e4059e962c
treeddfbff2f853fc24eb9f07189bffbd9bcb70ab148
parent18cb3fc91004ac9e551301748246aaa2a5a5dd04
Implement leaky relu op

Summary: Implemented leaky relu op as per: https://www.internalfb.com/tasks/?t=97492679

Test Plan:
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"

all tests pass, including new ones

Reviewed By: SS-JIA

Differential Revision: D30186225

fbshipit-source-id: fdb1f8f7b3a28b5504581822185c0475dcd53a3e
aten/src/ATen/native/vulkan/glsl/leaky_relu.glsl [new file with mode: 0644]
aten/src/ATen/native/vulkan/glsl/leaky_relu_.glsl [new file with mode: 0644]
aten/src/ATen/native/vulkan/ops/Clamp.cpp
aten/src/ATen/test/vulkan_api_test.cpp