diff options
Diffstat (limited to 'src/caffe/layers/eltwise_layer.cpp')
| -rw-r--r-- | src/caffe/layers/eltwise_layer.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/caffe/layers/eltwise_layer.cpp b/src/caffe/layers/eltwise_layer.cpp index 0c239f4a..bbc34449 100644 --- a/src/caffe/layers/eltwise_layer.cpp +++ b/src/caffe/layers/eltwise_layer.cpp @@ -163,5 +163,6 @@ STUB_GPU(EltwiseLayer); #endif INSTANTIATE_CLASS(EltwiseLayer); -REGISTER_LAYER_CLASS(ELTWISE, EltwiseLayer); +REGISTER_LAYER_CLASS(Eltwise); + } // namespace caffe |