10 #ifndef EIGEN_CXX11_TENSOR_TENSOR_LAYOUT_SWAP_H
11 #define EIGEN_CXX11_TENSOR_TENSOR_LAYOUT_SWAP_H
38 template<
typename XprType>
39 struct traits<TensorLayoutSwapOp<XprType> > :
public traits<XprType>
41 typedef typename XprType::Scalar Scalar;
42 typedef traits<XprType> XprTraits;
43 typedef typename packet_traits<Scalar>::type Packet;
44 typedef typename XprTraits::StorageKind StorageKind;
45 typedef typename XprTraits::Index Index;
46 typedef typename XprType::Nested Nested;
47 typedef typename remove_reference<Nested>::type _Nested;
48 static const int NumDimensions = traits<XprType>::NumDimensions;
49 static const int Layout = (traits<XprType>::Layout == ColMajor) ? RowMajor : ColMajor;
52 template<
typename XprType>
53 struct eval<TensorLayoutSwapOp<XprType>,
Eigen::Dense>
55 typedef const TensorLayoutSwapOp<XprType>& type;
58 template<
typename XprType>
59 struct nested<TensorLayoutSwapOp<XprType>, 1, typename eval<TensorLayoutSwapOp<XprType> >::type>
61 typedef TensorLayoutSwapOp<XprType> type;
68 template<
typename XprType>
69 class TensorLayoutSwapOp :
public TensorBase<TensorLayoutSwapOp<XprType>, WriteAccessors>
72 typedef typename Eigen::internal::traits<TensorLayoutSwapOp>::Scalar Scalar;
73 typedef typename Eigen::internal::traits<TensorLayoutSwapOp>::Packet Packet;
74 typedef typename Eigen::NumTraits<Scalar>::Real RealScalar;
75 typedef typename internal::remove_const<typename XprType::CoeffReturnType>::type CoeffReturnType;
76 typedef typename internal::remove_const<typename XprType::PacketReturnType>::type PacketReturnType;
77 typedef typename Eigen::internal::nested<TensorLayoutSwapOp>::type Nested;
78 typedef typename Eigen::internal::traits<TensorLayoutSwapOp>::StorageKind StorageKind;
79 typedef typename Eigen::internal::traits<TensorLayoutSwapOp>::Index Index;
81 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorLayoutSwapOp(
const XprType& expr)
85 const typename internal::remove_all<typename XprType::Nested>::type&
86 expression()
const {
return m_xpr; }
89 EIGEN_STRONG_INLINE TensorLayoutSwapOp& operator = (
const TensorLayoutSwapOp& other)
91 typedef TensorAssignOp<TensorLayoutSwapOp, const TensorLayoutSwapOp> Assign;
92 Assign assign(*
this, other);
93 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
97 template<
typename OtherDerived>
99 EIGEN_STRONG_INLINE TensorLayoutSwapOp& operator = (
const OtherDerived& other)
101 typedef TensorAssignOp<TensorLayoutSwapOp, const OtherDerived> Assign;
102 Assign assign(*
this, other);
103 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
108 typename XprType::Nested m_xpr;
113 template<
typename ArgType,
typename Device>
114 struct TensorEvaluator<const TensorLayoutSwapOp<ArgType>, Device>
116 typedef TensorLayoutSwapOp<ArgType> XprType;
117 typedef typename XprType::Index Index;
118 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
119 typedef DSizes<Index, NumDims> Dimensions;
122 IsAligned = TensorEvaluator<ArgType, Device>::IsAligned,
123 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
124 Layout = (
static_cast<int>(TensorEvaluator<ArgType, Device>::Layout) == static_cast<int>(ColMajor)) ? RowMajor : ColMajor,
128 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(
const XprType& op,
const Device& device)
129 : m_impl(op.expression(), device)
131 for(
int i = 0; i < NumDims; ++i) {
132 m_dimensions[i] = m_impl.dimensions()[NumDims-1-i];
136 typedef typename XprType::Scalar Scalar;
137 typedef typename XprType::CoeffReturnType CoeffReturnType;
138 typedef typename XprType::PacketReturnType PacketReturnType;
140 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
const Dimensions& dimensions()
const {
return m_dimensions; }
142 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
bool evalSubExprsIfNeeded(CoeffReturnType* data) {
143 return m_impl.evalSubExprsIfNeeded(data);
145 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
void cleanup() {
149 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index index)
const
151 return m_impl.coeff(index);
154 template<
int LoadMode>
155 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE PacketReturnType packet(Index index)
const
157 return m_impl.template packet<LoadMode>(index);
160 EIGEN_DEVICE_FUNC Scalar* data()
const {
return m_impl.data(); }
162 const TensorEvaluator<ArgType, Device>& impl()
const {
return m_impl; }
165 TensorEvaluator<ArgType, Device> m_impl;
166 Dimensions m_dimensions;
171 template<
typename ArgType,
typename Device>
172 struct TensorEvaluator<TensorLayoutSwapOp<ArgType>, Device>
173 :
public TensorEvaluator<const TensorLayoutSwapOp<ArgType>, Device>
175 typedef TensorEvaluator<const TensorLayoutSwapOp<ArgType>, Device> Base;
176 typedef TensorLayoutSwapOp<ArgType> XprType;
179 IsAligned = TensorEvaluator<ArgType, Device>::IsAligned,
180 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
181 Layout = (
static_cast<int>(TensorEvaluator<ArgType, Device>::Layout) == static_cast<int>(ColMajor)) ? RowMajor : ColMajor,
185 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(
const XprType& op,
const Device& device)
189 typedef typename XprType::Index Index;
190 typedef typename XprType::Scalar Scalar;
191 typedef typename XprType::CoeffReturnType CoeffReturnType;
192 typedef typename XprType::PacketReturnType PacketReturnType;
194 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType& coeffRef(Index index)
196 return this->m_impl.coeffRef(index);
198 template <
int StoreMode> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
199 void writePacket(Index index,
const PacketReturnType& x)
201 this->m_impl.template writePacket<StoreMode>(index, x);
207 #endif // EIGEN_CXX11_TENSOR_TENSOR_LAYOUT_SWAP_H
Namespace containing all symbols from the Eigen library.
Definition: CXX11Meta.h:13