diff options
author | Till Hoffmann <tillahoffmann@gmail.com> | 2016-04-01 17:58:24 +0100 |
---|---|---|
committer | Till Hoffmann <tillahoffmann@gmail.com> | 2016-04-01 17:58:24 +0100 |
commit | ffd770ce94b75202187635bf0e1e4d0006f4a015 (patch) | |
tree | 095fbbcd0523079bdbb189fa4bd08fda7731e1d4 /unsupported/Eigen/CXX11/src/Tensor/TensorBase.h | |
parent | 3cb0a237c195dd470ff5cd7a6e793b74d2d6815d (diff) |
Fixed CUDA signature.
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorBase.h')
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorBase.h | 26 |
1 files changed, 14 insertions, 12 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h b/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h index 65b969aaf..77b509f61 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h @@ -132,18 +132,6 @@ class TensorBase<Derived, ReadOnlyAccessors> digamma() const { return unaryExpr(internal::scalar_digamma_op<Scalar>()); } - - EIGEN_DEVICE_FUNC - EIGEN_STRONG_INLINE const TensorCwiseUnaryOp<internal::scalar_zeta_op<Scalar>, const Derived> - zeta() const { - return unaryExpr(internal::scalar_zeta_op<Scalar>()); - } - - EIGEN_DEVICE_FUNC - EIGEN_STRONG_INLINE const TensorCwiseUnaryOp<internal::scalar_polygamma_op<Scalar>, const Derived> - polygamma() const { - return unaryExpr(internal::scalar_polygamma_op<Scalar>()); - } EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp<internal::scalar_erf_op<Scalar>, const Derived> @@ -365,6 +353,20 @@ class TensorBase<Derived, ReadOnlyAccessors> igammac(const OtherDerived& other) const { return binaryExpr(other.derived(), internal::scalar_igammac_op<Scalar>()); } + + // zeta(x = this, q = other) + template<typename OtherDerived> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE + const TensorCwiseBinaryOp<internal::scalar_zeta_op<Scalar>, const Derived, const OtherDerived> + igammac(const OtherDerived& other) const { + return binaryExpr(other.derived(), internal::scalar_igammac_op<Scalar>()); + } + + // polygamma(n = this, x = other) + template<typename OtherDerived> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE + const TensorCwiseBinaryOp<internal::scalar_polygamma_op<Scalar>, const Derived, const OtherDerived> + igammac(const OtherDerived& other) const { + return binaryExpr(other.derived(), internal::scalar_igammac_op<Scalar>()); + } // comparisons and tests for Scalars EIGEN_DEVICE_FUNC |