Skip to content

P
Paddle-Lite

PaddlePaddle / Paddle-Lite

通知 331
Star 4
Fork 1
P
Paddle-Lite
提交 d77a180c 编写于 作者: E eclipsess
浏览文件
操作

code style

上级 4f8db678
显示空白变更内容
内联 并排
Showing with 14 addition and 9 deletion
src/operators/math/pool_2x2.cpp
浏览文件 @ d77a180c
...@@ -123,9 +123,12 @@ void Pool2x2Maxs2p0(vector<int> strides, vector<int> paddings, ...@@ -123,9 +123,12 @@ void Pool2x2Maxs2p0(vector<int> strides, vector<int> paddings,
if (_w2 != 0) { if (_w2 != 0) {
in_ptr1 = input_data + i * input_batch_stride + in_ptr1 = input_data + i * input_batch_stride +
c * input_channel_stride + ph * input_width+16 * w1 + 4 * w2; c * input_channel_stride + ph * input_width + 16 * w1 +
4 * w2;
in_ptr2 = in_ptr1 + input_width; in_ptr2 = in_ptr1 + input_width;
out_ptr = output_data + i * output_batch_stride + c * output_channel_stride + ph / 2 * output_width + 8 * w1 + 2 * w2; out_ptr = output_data + i * output_batch_stride +
c * output_channel_stride + ph / 2 * output_width + 8 * w1 +
2 * w2;
if (_w2 == 1) { if (_w2 == 1) {
*out_ptr = (*in_ptr1 > *in_ptr2) ? *in_ptr1 : *in_ptr2; *out_ptr = (*in_ptr1 > *in_ptr2) ? *in_ptr1 : *in_ptr2;
} else if (_w2 == 2) { } else if (_w2 == 2) {
...@@ -259,10 +262,12 @@ void Pool2x2Avgs2p0(vector<int> strides, vector<int> paddings, ...@@ -259,10 +262,12 @@ void Pool2x2Avgs2p0(vector<int> strides, vector<int> paddings,
if (_w2 != 0) { if (_w2 != 0) {
in_ptr1 = input_data + i * input_batch_stride + in_ptr1 = input_data + i * input_batch_stride +
c * input_channel_stride + ph * input_width + 16 * w1 + 4 * w2; c * input_channel_stride + ph * input_width + 16 * w1 +
4 * w2;
in_ptr2 = in_ptr1 + input_width; in_ptr2 = in_ptr1 + input_width;
out_ptr = output_data + i * output_batch_stride + out_ptr = output_data + i * output_batch_stride +
c * output_channel_stride + ph / 2 * output_width + 8 * w1 + 2 * w2; c * output_channel_stride + ph / 2 * output_width + 8 * w1 +
2 * w2;
if (_w2 == 1) { if (_w2 == 1) {
*out_ptr = 0.5 * (*in_ptr1 + *in_ptr2); *out_ptr = 0.5 * (*in_ptr1 + *in_ptr2);
} else if (_w2 == 2) { } else if (_w2 == 2) {
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或