Home
last modified time | relevance | path

Searched refs:FFT_FORWARD (Results 1 – 6 of 6) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_fft.cpp29 …plex<float>, 2, DataLayout> output = input.template fft<Eigen::BothParts, Eigen::FFT_FORWARD>(fft); in test_fft_2D_golden()
70 …d::complex<float>, 1, ColMajor> forward_output_both_parts = input.fft<BothParts, FFT_FORWARD>(fft); in test_fft_complex_input_golden()
73 Tensor<float, 1, ColMajor> forward_output_real_part = input.fft<RealPart, FFT_FORWARD>(fft); in test_fft_complex_input_golden()
76 Tensor<float, 1, ColMajor> forward_output_imag_part = input.fft<ImagPart, FFT_FORWARD>(fft); in test_fft_complex_input_golden()
127 …d::complex<float>, 1, ColMajor> forward_output_both_parts = input.fft<BothParts, FFT_FORWARD>(fft); in test_fft_real_input_golden()
130 Tensor<float, 1, ColMajor> forward_output_real_part = input.fft<RealPart, FFT_FORWARD>(fft); in test_fft_real_input_golden()
133 Tensor<float, 1, ColMajor> forward_output_imag_part = input.fft<ImagPart, FFT_FORWARD>(fft); in test_fft_real_input_golden()
219 if(FFTDirection == FFT_FORWARD) { in test_fft_real_input_energy()
234 test_fft_real_input_energy<ColMajor, float, true, Eigen::BothParts, FFT_FORWARD, 1>(); in test_cxx11_tensor_fft()
235 test_fft_real_input_energy<ColMajor, double, true, Eigen::BothParts, FFT_FORWARD, 1>(); in test_cxx11_tensor_fft()
[all …]
Dcxx11_tensor_ifft.cpp28 tensor_after_fft = tensor.template fft<Eigen::BothParts, Eigen::FFT_FORWARD>(fft); in test_1D_fft_ifft_invariant()
51 tensor_after_fft = tensor.template fft<Eigen::BothParts, Eigen::FFT_FORWARD>(fft); in test_2D_fft_ifft_invariant()
80 tensor_after_fft = tensor.template fft<Eigen::BothParts, Eigen::FFT_FORWARD>(fft); in test_3D_fft_ifft_invariant()
111 tensor_after_fft = tensor.template fft<Eigen::BothParts, Eigen::FFT_FORWARD>(fft); in test_sub_fft_ifft_invariant()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorFFT.h273 if (FFTDir == FFT_FORWARD && stride == 1) {
279 buf[offset] = (FFTDir == FFT_FORWARD) ? line_buf[j] : line_buf[j] * div_factor;
331 if(FFTDir == FFT_FORWARD) {
343 if(FFTDir == FFT_FORWARD) {
354 if(FFTDir == FFT_FORWARD) {
363 compute_1D_Butterfly<FFT_FORWARD>(a, m, log_len);
366 compute_1D_Butterfly<FFT_FORWARD>(b, m, log_len);
381 if(FFTDir == FFT_FORWARD) {
419 if (Dir == FFT_FORWARD) {
438 if (Dir == FFT_FORWARD) {
[all …]
DTensorForwardDeclarations.h83 FFT_FORWARD = 0, enumerator
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_fft_impl.h40 constexpr auto direction = Forward ? Eigen::FFT_FORWARD : Eigen::FFT_REVERSE; in EigenFftC2C()
91 input.template fft<Eigen::BothParts, Eigen::FFT_FORWARD>(axes); in EigenFftR2C()
/external/tensorflow/tensorflow/core/kernels/
Dfft_ops.cc134 Forward ? Eigen::FFT_FORWARD : Eigen::FFT_REVERSE; in DoFFT()
173 .template fft<Eigen::BothParts, Eigen::FFT_FORWARD>(axes); in DoFFT()