Function fft_inplace

src/backend/cpu/fft.cpp:79–109

template<typename T>
void fft_inplace(Array<T> &in, const int rank, const bool direction) {
    auto func = [=](Param<T> in, const af::dim4 iDataDims) {
        const af::dim4 idims = in.dims();

        auto t_dims   = computeDims(rank, idims);
        auto in_embed = computeDims(rank, iDataDims);

        const af::dim4 istrides = in.strides();

        using ctype_t = typename fftw_transform<T>::ctype_t;
        typename fftw_transform<T>::plan_t plan;

        fftw_transform<T> transform;

        int batch = 1;
        for (int i = rank; i < 4; i++) { batch *= idims[i]; }

        plan = transform.create(
            rank, t_dims.data(), batch, reinterpret_cast<ctype_t *>(in.get()),
            in_embed.data(), static_cast<int>(istrides[0]),
            static_cast<int>(istrides[rank]),
            reinterpret_cast<ctype_t *>(in.get()), in_embed.data(),
            static_cast<int>(istrides[0]), static_cast<int>(istrides[rank]),
            direction ? FFTW_FORWARD : FFTW_BACKWARD,
            FFTW_ESTIMATE);  // NOLINT(hicpp-signed-bitwise)

        transform.execute(plan);
        transform.destroy(plan);
    };
    getQueue().enqueue(func, in, in.getDataDims());
}

template<typename Tc, typename Tr>
Array<Tc> fft_r2c(const Array<Tr> &in, const int rank) {