Remove incorrect cast of gemm int arg to Dtype in BiasLayer
authorJeff Donahue <jeff.donahue@gmail.com>
Wed, 27 Jan 2016 07:09:27 +0000 (23:09 -0800)
committerJeff Donahue <jeff.donahue@gmail.com>
Wed, 27 Jan 2016 07:09:27 +0000 (23:09 -0800)
src/caffe/layers/bias_layer.cpp

index 0a786b5..4726a72 100644 (file)
@@ -80,7 +80,7 @@ void BiasLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& 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_;
   }