| 354 | |
| 355 | template <typename T> |
| 356 | SimpleTensor<T> ridft_2d(const SimpleTensor<T> &src, bool is_odd) |
| 357 | { |
| 358 | ARM_COMPUTE_ERROR_ON(src.num_channels() != 2); |
| 359 | constexpr FFTDirection direction = FFTDirection::Inverse; |
| 360 | |
| 361 | auto transposed = permute(src, PermutationVector(1U, 0U)); |
| 362 | auto first_pass = dft_1d_core(transposed, direction); |
| 363 | auto transposed_2 = permute(first_pass, PermutationVector(1U, 0U)); |
| 364 | auto dst = rdft_1d_core(transposed_2, direction, is_odd); |
| 365 | |
| 366 | const T scaling_factor = T(dst.shape()[0] * dst.shape()[1]); |
| 367 | scale(dst, scaling_factor); |
| 368 | return dst; |
| 369 | } |
| 370 | |
| 371 | template <typename T> |
| 372 | SimpleTensor<T> dft_2d(const SimpleTensor<T> &src, FFTDirection direction) |
no test coverage detected