From d1da472ab4bd56a75b255be4b1f95feb9b9ff15a Mon Sep 17 00:00:00 2001 From: zhaojiaying01 Date: Wed, 11 Jul 2018 15:31:03 +0800 Subject: [PATCH] Revert: accelerate with openmp --- src/operators/kernel/lrn_kernel.h | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/operators/kernel/lrn_kernel.h b/src/operators/kernel/lrn_kernel.h index 82c9f7530c..7327451a0a 100644 --- a/src/operators/kernel/lrn_kernel.h +++ b/src/operators/kernel/lrn_kernel.h @@ -12,20 +12,23 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. */ -#pragma once +#ifdef LRN_OP -#include #include "framework/operator.h" #include "operators/op_param.h" +#include + #ifdef __ARM_NEON -#include +#include "arm_neon.h" #include "operators/math/math_func_neon.h" #endif namespace paddle_mobile { namespace operators { +using namespace framework; + template struct LRNFunctor { void operator()(const framework::Tensor &input, framework::Tensor *out, int N, -- GitLab