diff options
| author | Jonathan L Long <jonlong@cs.berkeley.edu> | 2014-12-29 17:59:07 -0800 |
|---|---|---|
| committer | Jonathan L Long <jonlong@cs.berkeley.edu> | 2014-12-29 17:59:07 -0800 |
| commit | 11add5bca9f31dc08329aa220d8674b7832c513d (patch) | |
| tree | d955bd1970e88c3b04710d105aa24f1214050e50 /src/caffe | |
| parent | a03d4f75ebcf236fbeab3e4155a87ce8f3819e13 (diff) | |
| download | caffeonacl-11add5bca9f31dc08329aa220d8674b7832c513d.tar.gz caffeonacl-11add5bca9f31dc08329aa220d8674b7832c513d.tar.bz2 caffeonacl-11add5bca9f31dc08329aa220d8674b7832c513d.zip | |
remove SoftmaxLossLayer CPU_ONLY stubs, since there is no GPU version
Diffstat (limited to 'src/caffe')
| -rw-r--r-- | src/caffe/layers/softmax_loss_layer.cpp | 6 |
1 files changed, 1 insertions, 5 deletions
diff --git a/src/caffe/layers/softmax_loss_layer.cpp b/src/caffe/layers/softmax_loss_layer.cpp index 82c62f6a..db8dd8bb 100644 --- a/src/caffe/layers/softmax_loss_layer.cpp +++ b/src/caffe/layers/softmax_loss_layer.cpp @@ -85,11 +85,7 @@ void SoftmaxWithLossLayer<Dtype>::Backward_cpu(const vector<Blob<Dtype>*>& top, } } - -#ifdef CPU_ONLY -STUB_GPU(SoftmaxWithLossLayer); -#endif - INSTANTIATE_CLASS(SoftmaxWithLossLayer); REGISTER_LAYER_CLASS(SOFTMAX_LOSS, SoftmaxWithLossLayer); + } // namespace caffe |