Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 0f6a7a65 编写于 作者: T tensor-tang
浏览文件
操作

fix typo

上级 41d05339
隐藏空白更改
内联 并排
Showing with 2 addition and 1 deletion
paddle/gserver/layers/MKLDNNBatchNormLayer.cpp
浏览文件 @ 0f6a7a65
...@@ -119,7 +119,7 @@ void MKLDNNBatchNormLayer::reshape( ...@@ -119,7 +119,7 @@ void MKLDNNBatchNormLayer::reshape(
int& bs, int& ic, int& ih, int& iw, int oc, int& oh, int& ow) { int& bs, int& ic, int& ih, int& iw, int oc, int& oh, int& ow) {
reshapeInput(bs, ih, iw); reshapeInput(bs, ih, iw);
oh = ih; oh = ih;
ow = ow; ow = iw;
// ic_ and oc can not be changed // ic_ and oc can not be changed
CHECK_EQ(inputElemenCnt_ / bs / ih / iw, (size_t)ic) CHECK_EQ(inputElemenCnt_ / bs / ih / iw, (size_t)ic)
<< "Input channel can not be changed"; << "Input channel can not be changed";
......
paddle/gserver/tests/test_MKLDNN.cpp
浏览文件 @ 0f6a7a65
...@@ -269,6 +269,7 @@ void testBatchNormLayer(const testBatchNormDesc& pm) { ...@@ -269,6 +269,7 @@ void testBatchNormLayer(const testBatchNormDesc& pm) {
TEST(MKLDNNLayer, BatchNormLayer) { TEST(MKLDNNLayer, BatchNormLayer) {
testBatchNormLayer({4, 10, 6, 6}); testBatchNormLayer({4, 10, 6, 6});
testBatchNormLayer({16, 32, 16, 16}); testBatchNormLayer({16, 32, 16, 16});
testBatchNormLayer({4, 16, 8, 10});
} }
struct testImageDesc { struct testImageDesc {
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或