diff options
| author | Jeff Donahue <jeff.donahue@gmail.com> | 2017-04-13 13:26:16 -0700 |
|---|---|---|
| committer | Jeff Donahue <jeff.donahue@gmail.com> | 2017-04-13 13:29:28 -0700 |
| commit | 1c15d94f7da736945450e6ed321077f3045445b1 (patch) | |
| tree | f3b6bc915792ed7a5231ae949d0ca613ff6e1550 | |
| parent | ab3398832964c1ff1bf6b78501e4e43a11f282a1 (diff) | |
| download | caffe-1c15d94f7da736945450e6ed321077f3045445b1.tar.gz caffe-1c15d94f7da736945450e6ed321077f3045445b1.tar.bz2 caffe-1c15d94f7da736945450e6ed321077f3045445b1.zip | |
CPU BatchNormLayer: replace powx with sqr and sqrt
| -rw-r--r-- | src/caffe/layers/batch_norm_layer.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/caffe/layers/batch_norm_layer.cpp b/src/caffe/layers/batch_norm_layer.cpp index 0a08ed4c..c6a1d5b1 100644 --- a/src/caffe/layers/batch_norm_layer.cpp +++ b/src/caffe/layers/batch_norm_layer.cpp @@ -124,8 +124,8 @@ void BatchNormLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom, if (!use_global_stats_) { // compute variance using var(X) = E((X-EX)^2) - caffe_powx(top[0]->count(), top_data, Dtype(2), - temp_.mutable_cpu_data()); // (X-EX)^2 + caffe_sqr<Dtype>(top[0]->count(), top_data, + temp_.mutable_cpu_data()); // (X-EX)^2 caffe_cpu_gemv<Dtype>(CblasNoTrans, channels_ * num, spatial_dim, 1. / (num * spatial_dim), temp_.cpu_data(), spatial_sum_multiplier_.cpu_data(), 0., @@ -148,7 +148,7 @@ void BatchNormLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom, // normalize variance caffe_add_scalar(variance_.count(), eps_, variance_.mutable_cpu_data()); - caffe_powx(variance_.count(), variance_.cpu_data(), Dtype(0.5), + caffe_sqrt(variance_.count(), variance_.cpu_data(), variance_.mutable_cpu_data()); // replicate variance to input size |