diff --git a/src/caffe/layers/mkldnn_inner_product_layer.cpp b/src/caffe/layers/mkldnn_inner_product_layer.cpp index b62ec3e4e..42302c56c 100644 --- a/src/caffe/layers/mkldnn_inner_product_layer.cpp +++ b/src/caffe/layers/mkldnn_inner_product_layer.cpp @@ -355,11 +355,11 @@ void MKLDNNInnerProductLayer::InitInnerProductBwd(const vector ipBwdData_desc; shared_ptr ipBwdWeights_desc; - if (this->bias_term_) - ipBwdWeights_desc.reset(new inner_product_backward_weights::desc(init_bottom_md, init_weights_md + if (this->bias_term_) + ipBwdWeights_desc.reset(new inner_product_backward_weights::desc(init_bottom_md, init_weights_md , init_bias_md, init_top_md)); - else - ipBwdWeights_desc.reset(new inner_product_backward_weights::desc(init_bottom_md, init_weights_md + 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));