diff --git a/src/operators/kernel/lrn_kernel.h b/src/operators/kernel/lrn_kernel.h index 82c9f7530ca878d0dc547c248a2103cbf8d314dc..7327451a0aa21b7bcf9ae111f63c19f2b6bb2d3a 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,