diff options
author | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2015-10-22 16:54:21 -0700 |
---|---|---|
committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2015-10-22 16:54:21 -0700 |
commit | ac99b4924976cb2d06a1747cd86e792de60f16c3 (patch) | |
tree | 644d8a593e931cec09cbc2fc921c34a20aaabba0 /unsupported/Eigen/CXX11/src/Tensor/TensorBase.h | |
parent | 2dd944661380875b5536658185041acbd459a225 (diff) |
Added missing glue logic
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorBase.h')
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorBase.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h b/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h index 1b85f5ef5..b004fdd7d 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h @@ -334,6 +334,13 @@ class TensorBase<Derived, ReadOnlyAccessors> return TensorConvolutionOp<const Dimensions, const Derived, const KernelDerived>(derived(), kernel.derived(), dims); } + // Fourier transforms + template <int FFTDataType, int FFTDirection, typename FFT> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE + const TensorFFTOp<const FFT, const Derived, FFTDataType, FFTDirection> + fft(const FFT& fft) const { + return TensorFFTOp<const FFT, const Derived, FFTDataType, FFTDirection>(derived(), fft); + } + // Reductions. template <typename Dims> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorReductionOp<internal::SumReducer<CoeffReturnType>, const Dims, const Derived> |