From ab4ceee35bde3b8626b33feb280015f1a5a85354 Mon Sep 17 00:00:00 2001 From: xiebaiyuan Date: Sat, 18 Aug 2018 14:09:49 +0800 Subject: [PATCH] format code in prelu_kernel --- src/operators/kernel/arm/prelu_kernel.cpp | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/src/operators/kernel/arm/prelu_kernel.cpp b/src/operators/kernel/arm/prelu_kernel.cpp index 57fb0951e8..a6b2606a1b 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]; } -- GitLab