From: Yangqing Jia Date: Fri, 27 Sep 2013 17:02:20 +0000 (-0700) Subject: inner product bugfix X-Git-Tag: submit/tizen/20180823.020014~992 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=0484890049df42d4407877be19cd7c03f60aee0c;p=platform%2Fupstream%2Fcaffeonacl.git inner product bugfix --- diff --git a/src/caffe/layers/inner_product_layer.cpp b/src/caffe/layers/inner_product_layer.cpp index 115dba7..e5e8e52 100644 --- a/src/caffe/layers/inner_product_layer.cpp +++ b/src/caffe/layers/inner_product_layer.cpp @@ -73,7 +73,7 @@ Dtype InnerProductLayer::Backward_cpu(const vector*>& top, const Dtype* top_diff = top[0]->cpu_diff(); const Dtype* bottom_data = (*bottom)[0]->cpu_data(); // Gradient with respect to weight - caffe_cpu_gemm(CblasTrans, CblasNoTrans, K_, N_, M_, (Dtype)1., + caffe_cpu_gemm(CblasTrans, CblasNoTrans, N_, K_, M_, (Dtype)1., top_diff, bottom_data, (Dtype)0., this->blobs_[0]->mutable_cpu_diff()); if (biasterm_) { // Gradient with respect to bias @@ -112,7 +112,7 @@ Dtype InnerProductLayer::Backward_gpu(const vector*>& top, const Dtype* top_diff = top[0]->gpu_diff(); const Dtype* bottom_data = (*bottom)[0]->gpu_data(); // Gradient with respect to weight - caffe_gpu_gemm(CblasTrans, CblasNoTrans, K_, N_, M_, (Dtype)1., + caffe_gpu_gemm(CblasTrans, CblasNoTrans, N_, K_, M_, (Dtype)1., top_diff, bottom_data, (Dtype)0., this->blobs_[0]->mutable_gpu_diff()); if (biasterm_) { // Gradient with respect to bias