From 756816db583fc8562f882b3f22792427e2b59010 Mon Sep 17 00:00:00 2001 From: PikachuHy <2931408816@qq.com> Date: Tue, 8 May 2018 09:14:02 +0800 Subject: [PATCH] fix #202 --- src/caffe/layers/mkldnn_inner_product_layer.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/caffe/layers/mkldnn_inner_product_layer.cpp b/src/caffe/layers/mkldnn_inner_product_layer.cpp index b62ec3e4e..fd6d886d2 100644 --- a/src/caffe/layers/mkldnn_inner_product_layer.cpp +++ b/src/caffe/layers/mkldnn_inner_product_layer.cpp @@ -358,10 +358,10 @@ void MKLDNNInnerProductLayer::InitInnerProductBwd(const vectorbias_term_) ipBwdWeights_desc.reset(new inner_product_backward_weights::desc(init_bottom_md, init_weights_md , init_bias_md, init_top_md)); - else + else{ ipBwdWeights_desc.reset(new inner_product_backward_weights::desc(init_bottom_md, init_weights_md , init_top_md)); - + } ipBwdData_desc.reset(new inner_product_backward_data::desc(init_bottom_md, init_weights_md, init_top_md)); // ---- Determining engine to use -----------------------