| 150 | } |
| 151 | |
| 152 | void NEFFT1D::run() |
| 153 | { |
| 154 | ARM_COMPUTE_TRACE_EVENT(ARM_COMPUTE_PROF_CAT_CPU, ARM_COMPUTE_PROF_LVL_CPU, "NEFFT1D::run"); |
| 155 | MemoryGroupResourceScope scope_mg(_memory_group); |
| 156 | |
| 157 | NEScheduler::get().schedule(_digit_reverse_kernel.get(), (_axis == 0 ? Window::DimY : Window::DimZ)); |
| 158 | |
| 159 | for (unsigned int i = 0; i < _num_ffts; ++i) |
| 160 | { |
| 161 | NEScheduler::get().schedule(_fft_kernels[i].get(), (_axis == 0 ? Window::DimY : Window::DimX)); |
| 162 | } |
| 163 | |
| 164 | // Run output scaling |
| 165 | if (_run_scale) |
| 166 | { |
| 167 | NEScheduler::get().schedule(_scale_kernel.get(), Window::DimY); |
| 168 | } |
| 169 | } |
| 170 | } // namespace arm_compute |