diff options
| author |  Benoit Steiner <benoit.steiner.goog@gmail.com> | 2015-02-26 09:42:41 -0800 |
|---|---|---|
| committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2015-02-26 09:42:41 -0800 |
| commit | f41b1f1666e91dc674a42fed9c444c91f483133f (patch) | |
| tree | f81dfac2e8e4ec0c3038cb8a814de31a7dafb9b6 /unsupported/Eigen/CXX11/src/Tensor/TensorBase.h | |
| parent | 8e817b65d02e6f6c2d4a0d0085212db0abe6c485 (diff) | |
Added support for fast reciprocal square root computation.
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorBase.h')
| -rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorBase.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h b/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h index cfcf18e8e..13709b504 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h @@ -78,6 +78,12 @@ class TensorBase<Derived, ReadOnlyAccessors> } EIGEN_DEVICE_FUNC + EIGEN_STRONG_INLINE const TensorCwiseUnaryOp<internal::scalar_rsqrt_op<Scalar>, const Derived> + rsqrt() const { + return unaryExpr(internal::scalar_rsqrt_op<Scalar>()); + } + + EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp<internal::scalar_square_op<Scalar>, const Derived> square() const { return unaryExpr(internal::scalar_square_op<Scalar>()); |