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

Function ridft_2d

tests/validation/reference/DFT.cpp:356–369  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

354
355template <typename T>
356SimpleTensor<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
371template <typename T>
372SimpleTensor<T> dft_2d(const SimpleTensor<T> &src, FFTDirection direction)

Callers 2

conv2d_dftFunction · 0.85
DATA_TEST_CASEFunction · 0.85

Calls 6

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

Tested by

no test coverage detected