未验证 提交 43e399c5 编写于 作者: T Tao Luo 提交者: GitHub

Merge pull request #5518 from tensor-tang/ds2

fix typo
...@@ -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";
......
...@@ -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.
先完成此消息的编辑!
想要评论请 注册