diff options
author | Rowland Depp <depp.rowland@gmail.com> | 2014-02-11 21:41:01 -0800 |
---|---|---|
committer | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-03-21 13:52:36 -0700 |
commit | a8c9b66b7f62610d71a18c798d5eb7157d49420c (patch) | |
tree | 295704e0d2fa2d201abd3b12626d0739a777f040 /src/caffe/layers/loss_layer.cpp | |
parent | 1cf822e53bee3eeca5dbc3c08a1e95171688ea9a (diff) | |
download | caffeonacl-a8c9b66b7f62610d71a18c798d5eb7157d49420c.tar.gz caffeonacl-a8c9b66b7f62610d71a18c798d5eb7157d49420c.tar.bz2 caffeonacl-a8c9b66b7f62610d71a18c798d5eb7157d49420c.zip |
major refactoring allow coexistence of MKL and non-MKL cases
Diffstat (limited to 'src/caffe/layers/loss_layer.cpp')
-rw-r--r-- | src/caffe/layers/loss_layer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/caffe/layers/loss_layer.cpp b/src/caffe/layers/loss_layer.cpp index 3c0f15fb..ef0074d5 100644 --- a/src/caffe/layers/loss_layer.cpp +++ b/src/caffe/layers/loss_layer.cpp @@ -154,7 +154,7 @@ void EuclideanLossLayer<Dtype>::Backward_cpu(const vector<Blob<Dtype>*>& top, int count = (*bottom)[0]->count(); int num = (*bottom)[0]->num(); // Compute the gradient - caffe_axpby(count, Dtype(1) / num, difference_.cpu_data(), Dtype(0), + caffe_cpu_axpby(count, Dtype(1) / num, difference_.cpu_data(), Dtype(0), (*bottom)[0]->mutable_cpu_diff()); } |