Merge pull request #3605 from jeffdonahue/bias-gemm-type-fix

Remove incorrect cast of gemm int arg to Dtype in BiasLayer
This commit is contained in:
Jeff Donahue 2016-01-27 10:34:20 -08:00
Родитель dc831aa8f5 afcaf253da
Коммит 0acb338b9a
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -80,7 +80,7 @@ void BiasLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom,
} }
for (int n = 0; n < outer_dim_; ++n) { for (int n = 0; n < outer_dim_; ++n) {
caffe_cpu_gemm(CblasNoTrans, CblasNoTrans, bias_dim_, 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); bias_multiplier_.cpu_data(), Dtype(1), top_data);
top_data += dim_; top_data += dim_;
} }