quantization - manaully merged from caffe-0.16

[jacinto-ai/caffe-jacinto.git] / include / caffe / layers / neuron_layer.hpp

diff --git a/include/caffe/layers/neuron_layer.hpp b/include/caffe/layers/neuron_layer.hpp
index 0e9c6ccdaeeef08ef3a98fc146a793143a07b7ff..b070e9d52053aa5f7c0246621d4211a9e8c97ec3 100644 (file)
--- a/include/caffe/layers/neuron_layer.hpp
+++ b/include/caffe/layers/neuron_layer.hpp
@@ -6,6 +6,7 @@
 #include "caffe/blob.hpp"
 #include "caffe/layer.hpp"
 #include "caffe/proto/caffe.pb.h"
+#include "caffe/quantized_layer.hpp"
 
 namespace caffe {
 
@@ -16,10 +17,10 @@ namespace caffe {
  *        element.
  */
 template <typename Ftype, typename Btype>
-class NeuronLayer : public Layer<Ftype, Btype> {
+class NeuronLayer : public QuantizedLayer<Ftype, Btype> {
  public:
   explicit NeuronLayer(const LayerParameter& param)
-     : Layer<Ftype, Btype>(param) {}
+     : QuantizedLayer<Ftype, Btype>(param) {}
   virtual void Reshape(const vector<Blob*>& bottom,
       const vector<Blob*>& top);