Merge pull request #5518 from tensor-tang/ds2

luotao1 · web-flow · commit 43e399c5a631 · 2017-11-09T21:04:07.000+08:00
fix typo
diff --git a/paddle/gserver/layers/MKLDNNBatchNormLayer.cpp b/paddle/gserver/layers/MKLDNNBatchNormLayer.cpp
@@ -119,7 +119,7 @@ void MKLDNNBatchNormLayer::reshape(
     int& bs, int& ic, int& ih, int& iw, int oc, int& oh, int& ow) {
   reshapeInput(bs, ih, iw);
   oh = ih;
-  ow = ow;
+  ow = iw;
   // ic_ and oc can not be changed
   CHECK_EQ(inputElemenCnt_ / bs / ih / iw, (size_t)ic)
       << "Input channel can not be changed";
diff --git a/paddle/gserver/tests/test_MKLDNN.cpp b/paddle/gserver/tests/test_MKLDNN.cpp
@@ -269,6 +269,7 @@ void testBatchNormLayer(const testBatchNormDesc& pm) {
 TEST(MKLDNNLayer, BatchNormLayer) {
   testBatchNormLayer({4, 10, 6, 6});
   testBatchNormLayer({16, 32, 16, 16});
+  testBatchNormLayer({4, 16, 8, 10});
 }
 
 struct testImageDesc {

Original file line number	Diff line number	Diff line change
`@@ -269,6 +269,7 @@ void testBatchNormLayer(const testBatchNormDesc& pm) {`
`269`	`269`	`TEST(MKLDNNLayer, BatchNormLayer) {`
`270`	`270`	`testBatchNormLayer({4, 10, 6, 6});`
`271`	`271`	`testBatchNormLayer({16, 32, 16, 16});`
	`272`	`+ testBatchNormLayer({4, 16, 8, 10});`
`272`	`273`	`}`
`273`	`274`
`274`	`275`	`struct testImageDesc {`