GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-07-01 18:55:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added support for fp16 to the sigmoid functor.

Browse Source
This commit is contained in:
Benoit Steiner 2016-05-10 12:25:27 -07:00
parent 0eb69b7552
commit 661e710092
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -69,7 +69,7 @@ struct scalar_sigmoid_op {
template <typename Packet> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE template <typename Packet> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
Packet packetOp(const Packet& x) const { Packet packetOp(const Packet& x) const {
const Packet one = pset1<Packet>(1); const Packet one = pset1<Packet>(T(1));
return pdiv(one, padd(one, pexp(pnegate(x)))); return pdiv(one, padd(one, pexp(pnegate(x))));
} }
}; };