Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
OPTHREE
Paddle
提交
343b1a96
P
Paddle
项目概览
OPTHREE
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
343b1a96
编写于
12月 04, 2017
作者:
T
tensor-tang
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add mkldnn_lrn unit test
上级
54205c99
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
45 addition
and
0 deletion
+45
-0
paddle/gserver/tests/test_MKLDNN.cpp
paddle/gserver/tests/test_MKLDNN.cpp
+45
-0
未找到文件。
paddle/gserver/tests/test_MKLDNN.cpp
浏览文件 @
343b1a96
...
...
@@ -272,6 +272,51 @@ TEST(MKLDNNLayer, BatchNormLayer) {
testBatchNormLayer
({
4
,
16
,
8
,
10
});
}
struct
testLRNDesc
{
int
bs
,
ic
,
ih
,
iw
;
float
scale
,
pow
;
int
localSize
;
};
void
getMKLDNNLRNConfig
(
TestConfig
&
cfg
,
const
testLRNDesc
&
pm
)
{
cfg
.
layerConfig
.
set_type
(
"mkldnn_lrn"
);
cfg
.
layerConfig
.
set_active_type
(
"relu"
);
size_t
layerSize
=
pm
.
ic
*
pm
.
ih
*
pm
.
iw
;
cfg
.
inputDefs
.
push_back
({
INPUT_DATA
,
"layer_0"
,
layerSize
,
0
});
LayerInputConfig
*
input
=
cfg
.
layerConfig
.
add_inputs
();
NormConfig
*
norm
=
input
->
mutable_norm_conf
();
norm
->
set_channels
(
pm
.
ic
);
norm
->
set_size
(
pm
.
localSize
);
norm
->
set_scale
(
pm
.
scale
);
norm
->
set_pow
(
pm
.
pow
);
norm
->
set_blocked
(
0
);
norm
->
set_img_size
(
pm
.
iw
);
norm
->
set_img_size_y
(
pm
.
ih
);
norm
->
set_output_x
(
norm
->
img_size
());
norm
->
set_output_y
(
norm
->
img_size_y
());
cfg
.
layerConfig
.
set_size
(
layerSize
);
cfg
.
biasSize
=
0
;
}
void
testLRNLayer
(
const
testLRNDesc
&
pm
)
{
TestConfig
dnnConfig
;
getMKLDNNLRNConfig
(
dnnConfig
,
pm
);
// mkldnn_lrn <==> norm with cmrnorm-projection type
TestConfig
refConfig
=
dnnConfig
;
refConfig
.
layerConfig
.
set_type
(
"norm"
);
LayerInputConfig
*
input
=
refConfig
.
layerConfig
.
mutable_inputs
(
0
);
NormConfig
*
norm
=
input
->
mutable_norm_conf
();
norm
->
set_norm_type
(
"cmrnorm-projection"
);
norm
->
set_scale
(
norm
->
scale
()
/
norm
->
size
());
RUN_MKLDNN_TEST
(
dnnConfig
,
refConfig
,
pm
)
}
TEST
(
MKLDNNLayer
,
LRNLayer
)
{
testLRNLayer
({
4
,
10
,
12
,
12
,
0.001
f
,
0.75
f
,
5
});
testLRNLayer
({
2
,
32
,
6
,
6
,
0.001
f
,
0.75
f
,
5
});
testLRNLayer
({
4
,
16
,
8
,
10
,
0.01
f
,
0.5
f
,
5
});
}
struct
testImageDesc
{
int
bs
,
ic
,
ih
,
iw
;
};
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录