diff --git a/src/operators/kernel/arm/prelu_kernel.cpp b/src/operators/kernel/arm/prelu_kernel.cpp index 57fb0951e87c8e3797c7d0e05af3eecebb2e60ef..a6b2606a1b5bced4e0e3e429223af57cee316141 100644 --- a/src/operators/kernel/arm/prelu_kernel.cpp +++ b/src/operators/kernel/arm/prelu_kernel.cpp @@ -48,22 +48,19 @@ void PReluKernel::Compute(const PReluParam ¶m) const { int i = 0; int temp = 0; if (mode == "channel") { -#pragma omp parallel for - + #pragma omp parallel for for (i = 0; i < numel; i++) { temp = numel / (dim[0] * dim[1]); index = (i / temp) % dim[1]; o_ptr[i] = x_ptr[i] > 0 ? x_ptr[i] : alpha_ptr[index] * x_ptr[i]; } } else if (mode == "element") { -#pragma omp parallel for - + #pragma omp parallel for for (i = 0; i < numel; i++) { o_ptr[i] = x_ptr[i] > 0 ? x_ptr[i] : alpha_ptr[i] * x_ptr[i]; } } else { -#pragma omp parallel for - + #pragma omp parallel for for (i = 0; i < numel; i++) { o_ptr[i] = x_ptr[i] > 0 ? x_ptr[i] : alpha_ptr[0] * x_ptr[i]; }