diff options
author | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2015-05-26 09:37:23 -0700 |
---|---|---|
committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2015-05-26 09:37:23 -0700 |
commit | 466bcc589ecdbbba0d594fe350029efbd8836218 (patch) | |
tree | 8d6fd392156325f6e796a315d9219c53653cfa73 | |
parent | d457734a191a17ac46ffcf36f85abfd87d0dd943 (diff) |
Added a few missing EIGEN_DEVICE_FUNC statements
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorExpr.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorExpr.h b/unsupported/Eigen/CXX11/src/Tensor/TensorExpr.h index b66b3ec2c..194c68929 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorExpr.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorExpr.h @@ -276,16 +276,20 @@ class TensorSelectOp : public TensorBase<TensorSelectOp<IfXprType, ThenXprType, typedef typename Eigen::internal::traits<TensorSelectOp>::StorageKind StorageKind; typedef typename Eigen::internal::traits<TensorSelectOp>::Index Index; + EIGEN_DEVICE_FUNC TensorSelectOp(const IfXprType& a_condition, const ThenXprType& a_then, const ElseXprType& a_else) : m_condition(a_condition), m_then(a_then), m_else(a_else) { } + EIGEN_DEVICE_FUNC const IfXprType& ifExpression() const { return m_condition; } + EIGEN_DEVICE_FUNC const ThenXprType& thenExpression() const { return m_then; } + EIGEN_DEVICE_FUNC const ElseXprType& elseExpression() const { return m_else; } protected: |