Enable BFloat16 LeakyReLU and RReLU in CPU path (#61514)
authoryanbing-j <yanbing.jiang@intel.com>
Tue, 24 Aug 2021 15:32:33 +0000 (08:32 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Tue, 24 Aug 2021 15:34:56 +0000 (08:34 -0700)
commit33a163d886d1a7b236bc34e69d5a7415a133bc23
tree2b035f8fff63a90818df21c44eea77c8e4abd223
parent2ca2761f3c448e58d587b440d459d328ae8efc95
Enable BFloat16 LeakyReLU and RReLU in CPU path (#61514)

Summary:
Enable and optimize BFloat16 LeakyReLU and RReLU in CPU path.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/61514

Reviewed By: ejguan

Differential Revision: D30257612

Pulled By: VitalyFedyunin

fbshipit-source-id: 8cc0d1faacd02dcc9827af724a86d95b6952748f
aten/src/ATen/native/Activation.cpp
aten/src/ATen/native/cpu/Activation.cpp
test/cpp/api/functional.cpp
test/cpp/api/modules.cpp
test/test_autograd.py
torch/testing/_internal/common_methods_invocations.py