MCPcopy Create free account
hub / github.com/ARM-software/ComputeLibrary / rdft_2d

Function rdft_2d

tests/validation/reference/DFT.cpp:344–353  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

342
343template <typename T>
344SimpleTensor<T> rdft_2d(const SimpleTensor<T> &src)
345{
346 ARM_COMPUTE_ERROR_ON(src.num_channels() != 1);
347 constexpr FFTDirection direction = FFTDirection::Forward;
348
349 auto first_pass = rdft_1d_core(src, direction, false);
350 auto transposed = permute(first_pass, PermutationVector(1U, 0U));
351 auto second_pass = dft_1d_core(transposed, direction);
352 return permute(second_pass, PermutationVector(1U, 0U));
353}
354
355template <typename T>
356SimpleTensor<T> ridft_2d(const SimpleTensor<T> &src, bool is_odd)

Callers 2

conv2d_dftFunction · 0.85
DATA_TEST_CASEFunction · 0.85

Calls 4

rdft_1d_coreFunction · 0.85
dft_1d_coreFunction · 0.85
permuteFunction · 0.70
num_channelsMethod · 0.45

Tested by

no test coverage detected