From 466bcc589ecdbbba0d594fe350029efbd8836218 Mon Sep 17 00:00:00 2001 From: Benoit Steiner Date: Tue, 26 May 2015 09:37:23 -0700 Subject: [PATCH] Added a few missing EIGEN_DEVICE_FUNC statements --- unsupported/Eigen/CXX11/src/Tensor/TensorExpr.h | 4 ++++ 1 file changed, 4 insertions(+) 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::StorageKind StorageKind; typedef typename Eigen::internal::traits::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: