Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
as350144
Mace
提交
8bd4b2be
Mace
项目概览
as350144
/
Mace
与 Fork 源项目一致
Fork自
Xiaomi / Mace
通知
2
Star
1
Fork
1
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
Mace
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
8bd4b2be
编写于
11月 09, 2017
作者:
L
liuqi
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Change the pooling functor init logic.
上级
47eece0b
变更
4
显示空白变更内容
内联
并排
Showing
4 changed file
with
51 addition
and
18 deletion
+51
-18
mace/kernels/neon/pooling_neon.cc
mace/kernels/neon/pooling_neon.cc
+12
-8
mace/kernels/pooling.h
mace/kernels/pooling.h
+18
-5
mace/ops/BUILD
mace/ops/BUILD
+16
-0
mace/ops/pooling.h
mace/ops/pooling.h
+5
-5
未找到文件。
mace/kernels/neon/pooling_neon.cc
浏览文件 @
8bd4b2be
...
...
@@ -3,8 +3,6 @@
//
#include "mace/kernels/pooling.h"
#include <arm_neon.h>
#include "mace/kernels/conv_pool_2d_util.h"
namespace
mace
{
namespace
kernels
{
...
...
@@ -61,9 +59,15 @@ void PoolingFunctor<DeviceType::NEON, float>::operator()(
const
index_t
*
input_shape
,
float
*
output
,
const
index_t
*
output_shape
)
{
int
paddings
[
2
];
std
::
vector
<
index_t
>
filter_shape
=
{
input_shape
[
1
],
input_shape
[
0
],
kernels_
[
0
],
kernels_
[
1
]};
kernels
::
CalPaddingSize
(
input_shape
,
filter_shape
.
data
(),
this
->
dilations_
,
strides_
,
this
->
padding_
,
paddings
);
#ifdef __COPY_MAKE_PADDING
Tensor
padded_input
;
ConstructInputWithPadding
(
input
,
input_shape
,
paddings
_
,
&
padded_input
);
ConstructInputWithPadding
(
input
,
input_shape
,
paddings
,
&
padded_input
);
input
=
padded_input
.
data
<
float
>
();
input_shape
=
padded_input
.
shape
().
data
();
#endif
...
...
@@ -76,14 +80,14 @@ void PoolingFunctor<DeviceType::NEON, float>::operator()(
PoolingMaxNeonK2x2S2x2Padded
(
input
,
input_shape
,
output
,
output_shape
);
#else
PoolingMaxNeonK2x2S2x2
(
input
,
input_shape
,
output
,
output_shape
,
paddings
_
);
paddings
);
#endif
}
else
{
// AVG_POOL_2x2s2x2
#ifdef __COPY_MAKE_PADDING
PoolingAvgNeonK2x2S2x2Padded
(
input
,
input_shape
,
output
,
output_shape
);
#else
PoolingAvgNeonK2x2S2x2
(
input
,
input_shape
,
output
,
output_shape
,
paddings
_
);
paddings
);
#endif
}
}
else
if
(
kernels_
[
0
]
==
3
&&
kernels_
[
1
]
==
3
&&
strides_
[
0
]
==
2
&&
...
...
@@ -94,19 +98,19 @@ void PoolingFunctor<DeviceType::NEON, float>::operator()(
PoolingMaxNeonK3x3S2x2Padded
(
input
,
input_shape
,
output
,
output_shape
);
#else
PoolingMaxNeonK3x3S2x2
(
input
,
input_shape
,
output
,
output_shape
,
paddings
_
);
paddings
);
#endif
}
else
{
// AVG_POOL_3x3s2x2
#ifdef __COPY_MAKE_PADDING
PoolingAvgNeonK3x3S2x2Padded
(
input
,
input_shape
,
output
,
output_shape
);
#else
PoolingAvgNeonK3x3S2x2
(
input
,
input_shape
,
output
,
output_shape
,
paddings
_
);
paddings
);
#endif
}
}
else
{
// not implement yet
PoolingFunctor
<
DeviceType
::
CPU
,
float
>
(
pooling_type_
,
kernels_
,
strides_
,
padding
s
_
,
dilations_
)(
padding_
,
dilations_
)(
input
,
input_shape
,
output
,
output_shape
);
}
}
...
...
mace/kernels/pooling.h
浏览文件 @
8bd4b2be
...
...
@@ -7,6 +7,7 @@
#include <limits>
#include "mace/core/tensor.h"
#include "mace/kernels/conv_pool_2d_util.h"
namespace
mace
{
...
...
@@ -22,12 +23,12 @@ struct PoolingFunctor {
PoolingFunctor
(
const
PoolingType
pooling_type
,
const
int
*
kernels
,
const
int
*
strides
,
const
int
*
paddings
,
const
Padding
padding
,
const
int
*
dilations
)
:
pooling_type_
(
pooling_type
),
kernels_
(
kernels
),
strides_
(
strides
),
padding
s_
(
paddings
),
padding
_
(
padding
),
dilations_
(
dilations
)
{}
void
operator
()(
const
T
*
input
,
...
...
@@ -54,9 +55,14 @@ struct PoolingFunctor {
int
dilation_h
=
dilations_
[
0
];
int
dilation_w
=
dilations_
[
1
];
int
paddings
[
2
];
std
::
vector
<
index_t
>
filter_shape
=
{
input_shape
[
1
],
input_shape
[
0
],
kernels_
[
0
],
kernels_
[
1
]};
kernels
::
CalPaddingSize
(
input_shape
,
filter_shape
.
data
(),
this
->
dilations_
,
strides_
,
this
->
padding_
,
paddings
);
// The left-upper most offset of the padded input
int
padded_h_start
=
0
-
paddings
_
[
0
]
/
2
;
int
padded_w_start
=
0
-
paddings
_
[
1
]
/
2
;
int
padded_h_start
=
0
-
paddings
[
0
]
/
2
;
int
padded_w_start
=
0
-
paddings
[
1
]
/
2
;
if
(
pooling_type_
==
MAX
)
{
#pragma omp parallel for collapse(2)
...
...
@@ -116,7 +122,7 @@ struct PoolingFunctor {
const
PoolingType
pooling_type_
;
const
int
*
kernels_
;
const
int
*
strides_
;
const
int
*
paddings
_
;
const
Padding
padding
_
;
const
int
*
dilations_
;
};
...
...
@@ -127,6 +133,13 @@ void PoolingFunctor<DeviceType::NEON, float>::operator()(
float
*
output
,
const
index_t
*
output_shape
);
template
<
>
void
PoolingFunctor
<
DeviceType
::
OPENCL
,
float
>::
operator
()(
const
float
*
input
,
const
index_t
*
input_shape
,
float
*
output
,
const
index_t
*
output_shape
);
}
// namespace kernels
}
// namespace mace
...
...
mace/ops/BUILD
浏览文件 @
8bd4b2be
...
...
@@ -62,6 +62,22 @@ cc_test(
],
)
cc_test
(
name
=
"pooling_test"
,
testonly
=
1
,
srcs
=
glob
(
[
"pooling_test.cc"
],
),
copts
=
[
"-std=c++11"
],
linkopts
=
[
"-fopenmp"
]
+
if_android
([
"-ldl"
]),
linkstatic
=
1
,
deps
=
[
":ops"
,
":test"
,
"@gtest//:gtest_main"
,
],
)
cc_test
(
name
=
"ops_benchmark"
,
testonly
=
1
,
...
...
mace/ops/pooling.h
浏览文件 @
8bd4b2be
...
...
@@ -19,7 +19,9 @@ class PoolingOp : public ConvPool2dOpBase<D, T> {
kernels_
(
OperatorBase
::
GetRepeatedArgument
<
int
>
(
"kernels"
)),
pooling_type_
(
static_cast
<
PoolingType
>
(
OperatorBase
::
GetSingleArgument
<
int
>
(
"pooling_type"
,
static_cast
<
int
>
(
AVG
)))){};
"pooling_type"
,
static_cast
<
int
>
(
AVG
)))),
functor_
(
pooling_type_
,
kernels_
.
data
(),
ConvPool2dOpBase
::
strides_
.
data
(),
ConvPool2dOpBase
::
padding_
,
ConvPool2dOpBase
::
dilations_
.
data
()){};
bool
Run
()
override
{
const
Tensor
*
input
=
this
->
Input
(
INPUT
);
...
...
@@ -40,10 +42,7 @@ class PoolingOp : public ConvPool2dOpBase<D, T> {
paddings
.
data
());
output
->
Resize
(
output_shape
);
auto
pooling_func
=
kernels
::
PoolingFunctor
<
D
,
T
>
(
pooling_type_
,
kernels_
.
data
(),
this
->
strides_
.
data
(),
paddings
.
data
(),
this
->
dilations_
.
data
());
pooling_func
(
input
->
data
<
float
>
(),
input
->
shape
().
data
(),
functor_
(
input
->
data
<
float
>
(),
input
->
shape
().
data
(),
output
->
mutable_data
<
float
>
(),
output
->
shape
().
data
());
return
true
;
};
...
...
@@ -51,6 +50,7 @@ class PoolingOp : public ConvPool2dOpBase<D, T> {
protected:
std
::
vector
<
int
>
kernels_
;
PoolingType
pooling_type_
;
kernels
::
PoolingFunctor
<
D
,
T
>
functor_
;
OP_INPUT_TAGS
(
INPUT
);
OP_OUTPUT_TAGS
(
OUTPUT
);
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录