diff options
author | android-build-team Robot <android-build-team-robot@google.com> | 2017-07-27 07:28:16 +0000 |
---|---|---|
committer | android-build-team Robot <android-build-team-robot@google.com> | 2017-07-27 07:28:16 +0000 |
commit | 5aa089d66ba903f93eba825f7a43376f734de697 (patch) | |
tree | 3ff1e643f70aad062b94a29f53ea8f422ed59e18 | |
parent | 24ca96fb9f8db1430a3b31b75443134a7577837c (diff) | |
parent | e241ae53c5a2c986c6077fac054d74f648079ea6 (diff) | |
download | eigen-5aa089d66ba903f93eba825f7a43376f734de697.tar.gz |
release-request-94878621-f29d-487c-a2aa-213f2be9d20c-for-git_oc-mr1-release-4222039 snap-temp-L91400000086509174
Change-Id: I35c65e626c4d3a8e5ccb170c550a45829187d177
-rw-r--r-- | Eigen/src/Core/functors/UnaryFunctors.h | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/Eigen/src/Core/functors/UnaryFunctors.h b/Eigen/src/Core/functors/UnaryFunctors.h index 2e6a00ffd..581a3c93a 100644 --- a/Eigen/src/Core/functors/UnaryFunctors.h +++ b/Eigen/src/Core/functors/UnaryFunctors.h @@ -14,6 +14,37 @@ namespace Eigen { namespace internal { +// Copied from unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h +// TODO: remove or update after upstream +/** \internal + * \brief Template functor to compute the sigmoid of a scalar + * \sa class CwiseUnaryOp, ArrayBase::sigmoid() + */ +template <typename T> +struct scalar_sigmoid_op { + EIGEN_EMPTY_STRUCT_CTOR(scalar_sigmoid_op) + EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE T operator()(const T& x) const { + const T one = T(1); + return one / (one + numext::exp(-x)); + } + + template <typename Packet> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE + Packet packetOp(const Packet& x) const { + const Packet one = pset1<Packet>(T(1)); + return pdiv(one, padd(one, pexp(pnegate(x)))); + } +}; + +template <typename T> +struct functor_traits<scalar_sigmoid_op<T> > { + enum { + Cost = NumTraits<T>::AddCost * 2 + NumTraits<T>::MulCost * 6, + PacketAccess = packet_traits<T>::HasAdd && packet_traits<T>::HasDiv && + packet_traits<T>::HasNegate && packet_traits<T>::HasExp + }; +}; + + /** \internal * \brief Template functor to compute the opposite of a scalar * |