From afcaf253daa942821250db3f9a6afbe1d955bdf1 Mon Sep 17 00:00:00 2001 From: Jeff Donahue Date: Tue, 26 Jan 2016 23:09:27 -0800 Subject: [PATCH] Remove incorrect cast of gemm int arg to Dtype in BiasLayer --- src/caffe/layers/bias_layer.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/caffe/layers/bias_layer.cpp b/src/caffe/layers/bias_layer.cpp index 0a786b5..4726a72 100644 --- a/src/caffe/layers/bias_layer.cpp +++ b/src/caffe/layers/bias_layer.cpp @@ -80,7 +80,7 @@ void BiasLayer::Forward_cpu(const vector*>& bottom, } for (int n = 0; n < outer_dim_; ++n) { caffe_cpu_gemm(CblasNoTrans, CblasNoTrans, bias_dim_, - inner_dim_, Dtype(1), Dtype(1), bias_data, + inner_dim_, 1, Dtype(1), bias_data, bias_multiplier_.cpu_data(), Dtype(1), top_data); top_data += dim_; } -- 2.7.4