11 #ifndef EIGEN_CXX11_TENSOR_TENSOR_REVERSE_H 12 #define EIGEN_CXX11_TENSOR_TENSOR_REVERSE_H 22 template<
typename ReverseDimensions,
typename XprType>
23 struct traits<TensorReverseOp<ReverseDimensions,
24 XprType> > :
public traits<XprType>
26 typedef typename XprType::Scalar Scalar;
27 typedef traits<XprType> XprTraits;
28 typedef typename XprTraits::StorageKind StorageKind;
29 typedef typename XprTraits::Index Index;
30 typedef typename XprType::Nested Nested;
31 typedef typename remove_reference<Nested>::type _Nested;
32 static const int NumDimensions = XprTraits::NumDimensions;
33 static const int Layout = XprTraits::Layout;
36 template<
typename ReverseDimensions,
typename XprType>
37 struct eval<TensorReverseOp<ReverseDimensions, XprType>,
Eigen::Dense>
39 typedef const TensorReverseOp<ReverseDimensions, XprType>& type;
42 template<
typename ReverseDimensions,
typename XprType>
43 struct nested<TensorReverseOp<ReverseDimensions, XprType>, 1,
44 typename eval<TensorReverseOp<ReverseDimensions, XprType> >::type>
46 typedef TensorReverseOp<ReverseDimensions, XprType> type;
51 template<
typename ReverseDimensions,
typename XprType>
52 class TensorReverseOp :
public TensorBase<TensorReverseOp<ReverseDimensions,
53 XprType>, WriteAccessors>
56 typedef typename Eigen::internal::traits<TensorReverseOp>::Scalar Scalar;
57 typedef typename Eigen::NumTraits<Scalar>::Real RealScalar;
58 typedef typename XprType::CoeffReturnType CoeffReturnType;
59 typedef typename Eigen::internal::nested<TensorReverseOp>::type Nested;
60 typedef typename Eigen::internal::traits<TensorReverseOp>::StorageKind
62 typedef typename Eigen::internal::traits<TensorReverseOp>::Index Index;
64 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorReverseOp(
65 const XprType& expr,
const ReverseDimensions& reverse_dims)
66 : m_xpr(expr), m_reverse_dims(reverse_dims) { }
69 const ReverseDimensions& reverse()
const {
return m_reverse_dims; }
72 const typename internal::remove_all<typename XprType::Nested>::type&
73 expression()
const {
return m_xpr; }
76 EIGEN_STRONG_INLINE TensorReverseOp& operator = (
const TensorReverseOp& other)
78 typedef TensorAssignOp<TensorReverseOp, const TensorReverseOp> Assign;
79 Assign assign(*
this, other);
80 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
84 template<
typename OtherDerived>
86 EIGEN_STRONG_INLINE TensorReverseOp& operator = (
const OtherDerived& other)
88 typedef TensorAssignOp<TensorReverseOp, const OtherDerived> Assign;
89 Assign assign(*
this, other);
90 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
95 typename XprType::Nested m_xpr;
96 const ReverseDimensions m_reverse_dims;
100 template<
typename ReverseDimensions,
typename ArgType,
typename Device>
101 struct TensorEvaluator<const TensorReverseOp<ReverseDimensions, ArgType>, Device>
103 typedef TensorReverseOp<ReverseDimensions, ArgType> XprType;
104 typedef typename XprType::Index Index;
105 static const int NumDims = internal::array_size<ReverseDimensions>::value;
106 typedef DSizes<Index, NumDims> Dimensions;
107 typedef typename XprType::Scalar Scalar;
108 typedef typename XprType::CoeffReturnType CoeffReturnType;
109 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
110 static const int PacketSize = internal::unpacket_traits<PacketReturnType>::size;
114 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
115 Layout = TensorEvaluator<ArgType, Device>::Layout,
120 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(
const XprType& op,
121 const Device& device)
122 : m_impl(op.expression(), device), m_reverse(op.reverse())
125 EIGEN_STATIC_ASSERT((NumDims > 0), YOU_MADE_A_PROGRAMMING_MISTAKE);
128 m_dimensions = m_impl.dimensions();
129 if (static_cast<int>(Layout) == static_cast<int>(ColMajor)) {
131 for (
int i = 1; i < NumDims; ++i) {
132 m_strides[i] = m_strides[i-1] * m_dimensions[i-1];
135 m_strides[NumDims-1] = 1;
136 for (
int i = NumDims - 2; i >= 0; --i) {
137 m_strides[i] = m_strides[i+1] * m_dimensions[i+1];
142 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
143 const Dimensions& dimensions()
const {
return m_dimensions; }
145 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
bool evalSubExprsIfNeeded(Scalar*) {
146 m_impl.evalSubExprsIfNeeded(NULL);
149 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
void cleanup() {
153 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index reverseIndex(
155 eigen_assert(index < dimensions().TotalSize());
156 Index inputIndex = 0;
157 if (static_cast<int>(Layout) == static_cast<int>(ColMajor)) {
158 for (
int i = NumDims - 1; i > 0; --i) {
159 Index idx = index / m_strides[i];
160 index -= idx * m_strides[i];
162 idx = m_dimensions[i] - idx - 1;
164 inputIndex += idx * m_strides[i] ;
167 inputIndex += (m_dimensions[0] - index - 1);
172 for (
int i = 0; i < NumDims - 1; ++i) {
173 Index idx = index / m_strides[i];
174 index -= idx * m_strides[i];
176 idx = m_dimensions[i] - idx - 1;
178 inputIndex += idx * m_strides[i] ;
180 if (m_reverse[NumDims-1]) {
181 inputIndex += (m_dimensions[NumDims-1] - index - 1);
189 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(
191 return m_impl.coeff(reverseIndex(index));
194 template<
int LoadMode>
195 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
196 PacketReturnType packet(Index index)
const 198 EIGEN_STATIC_ASSERT((PacketSize > 1), YOU_MADE_A_PROGRAMMING_MISTAKE)
199 eigen_assert(index+PacketSize-1 < dimensions().TotalSize());
203 EIGEN_ALIGN_MAX
typename internal::remove_const<CoeffReturnType>::type
205 for (
int i = 0; i < PacketSize; ++i) {
206 values[i] = coeff(index+i);
208 PacketReturnType rslt = internal::pload<PacketReturnType>(values);
212 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost costPerCoeff(
bool vectorized)
const {
213 double compute_cost = NumDims * (2 * TensorOpCost::AddCost<Index>() +
214 2 * TensorOpCost::MulCost<Index>() +
215 TensorOpCost::DivCost<Index>());
216 for (
int i = 0; i < NumDims; ++i) {
218 compute_cost += 2 * TensorOpCost::AddCost<Index>();
221 return m_impl.costPerCoeff(vectorized) +
222 TensorOpCost(0, 0, compute_cost,
false , PacketSize);
225 EIGEN_DEVICE_FUNC Scalar* data()
const {
return NULL; }
228 Dimensions m_dimensions;
229 array<Index, NumDims> m_strides;
230 TensorEvaluator<ArgType, Device> m_impl;
231 ReverseDimensions m_reverse;
236 template <
typename ReverseDimensions,
typename ArgType,
typename Device>
237 struct TensorEvaluator<TensorReverseOp<ReverseDimensions, ArgType>, Device>
238 :
public TensorEvaluator<const TensorReverseOp<ReverseDimensions, ArgType>,
240 typedef TensorEvaluator<const TensorReverseOp<ReverseDimensions, ArgType>,
242 typedef TensorReverseOp<ReverseDimensions, ArgType> XprType;
243 typedef typename XprType::Index Index;
244 static const int NumDims = internal::array_size<ReverseDimensions>::value;
245 typedef DSizes<Index, NumDims> Dimensions;
249 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
250 Layout = TensorEvaluator<ArgType, Device>::Layout,
254 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(
const XprType& op,
255 const Device& device)
256 : Base(op, device) {}
258 typedef typename XprType::Scalar Scalar;
259 typedef typename XprType::CoeffReturnType CoeffReturnType;
260 typedef typename PacketType<CoeffReturnType, Device>::type PacketReturnType;
261 static const int PacketSize = internal::unpacket_traits<PacketReturnType>::size;
263 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
264 const Dimensions& dimensions()
const {
return this->m_dimensions; }
266 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Scalar& coeffRef(Index index) {
267 return this->m_impl.coeffRef(this->reverseIndex(index));
270 template <
int StoreMode> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
271 void writePacket(Index index,
const PacketReturnType& x) {
272 EIGEN_STATIC_ASSERT((PacketSize > 1), YOU_MADE_A_PROGRAMMING_MISTAKE)
273 eigen_assert(index+PacketSize-1 < dimensions().TotalSize());
276 EIGEN_ALIGN_MAX CoeffReturnType values[PacketSize];
277 internal::pstore<CoeffReturnType, PacketReturnType>(values, x);
278 for (
int i = 0; i < PacketSize; ++i) {
279 this->coeffRef(index+i) = values[i];
288 #endif // EIGEN_CXX11_TENSOR_TENSOR_REVERSE_H Namespace containing all symbols from the Eigen library.
Definition: AdolcForward:45