pool_3x3.h 1.3 KB
Newer Older
W
wangliu 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14
/* Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
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. */

L
liuruilong 已提交
15 16
#ifdef POOL_OP

17
#pragma once
W
wangliu 已提交
18 19
#include <algorithm>
#include <vector>
W
wangliu 已提交
20
#include "framework/tensor.h"
L
liuruilong 已提交
21
#ifdef __ARM_NEON
22
#include <arm_neon.h>
朔-望's avatar
朔-望 已提交
23
#endif  // __ARM_NEON
24

W
wangliu 已提交
25 26 27 28 29
namespace paddle_mobile {
namespace operators {
namespace math {
using framework::Tensor;
using std::vector;
W
wangliu 已提交
30 31
void Pool3x3Avgs1p1(const Tensor *input, Tensor *output);
void Pool3x3Maxs1p1(const Tensor *input, Tensor *output);
W
wangliu 已提交
32 33 34 35 36 37 38 39
void Pool3x3Max(vector<int> strides, vector<int> paddings, const Tensor *input,
                Tensor *output);

void Pool3x3Avg(vector<int> strides, vector<int> paddings, const Tensor *in_x,
                Tensor *out);
}  // namespace math
}  // namespace operators
}  // namespace paddle_mobile
L
liuruilong 已提交
40 41

#endif