diff --git a/src/caffe/layers/mkldnn_inner_product_layer.cpp b/src/caffe/layers/mkldnn_inner_product_layer.cpp index 139ac4dc3..0a879a621 100644 --- a/src/caffe/layers/mkldnn_inner_product_layer.cpp +++ b/src/caffe/layers/mkldnn_inner_product_layer.cpp @@ -354,10 +354,13 @@ void MKLDNNInnerProductLayer::InitInnerProductFwd(const vector 1) #endif for(int i=0; iscale_params_[i] == 0.0) - scale_bias[i] = 1.0; - else - scale_bias[i] = this->scale_in_[0] * this->scale_params_[i]; + if (this->scale_params_[i] == 0.0){ + scale_bias[i] = 1.0; + } + else{ + scale_bias[i] = this->scale_in_[0] * this->scale_params_[i]; + } + } } else{ scale_bias[0] = 1.0f;