GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-05-18 23:57:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added support for fp16 to the sigmoid function

Browse Source
This commit is contained in:
Benoit Steiner 2016-04-14 10:08:33 -07:00
parent f6003f0873
commit c7167fee0e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h
View File

@ -64,7 +64,7 @@ struct scalar_sigmoid_op {
EIGEN_EMPTY_STRUCT_CTOR(scalar_sigmoid_op) EIGEN_EMPTY_STRUCT_CTOR(scalar_sigmoid_op)
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE T operator()(const T& x) const { EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE T operator()(const T& x) const {
const T one = T(1); const T one = T(1);
return one / (one + std::exp(-x)); return one / (one + numext::exp(-x));
} }
template <typename Packet> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE template <typename Packet> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
@ -799,7 +799,7 @@ class GaussianGenerator {
T offset = coordinates[i] - m_means[i]; T offset = coordinates[i] - m_means[i];
tmp += offset * offset / m_two_sigmas[i]; tmp += offset * offset / m_two_sigmas[i];
} }
return std::exp(-tmp); return numext::exp(-tmp);
} }
private: private: