src/caffeine/neuron_layer.cpp

   1 #include "caffeine/layer.hpp"
   2 #include "caffeine/vision_layers.hpp"
   3
   4 namespace caffeine {
   5
   6 template <typename Dtype>
   7 void NeuronLayer<Dtype>::SetUp(const vector<Blob<Dtype>*>& bottom,
   8       vector<Blob<Dtype>*>* top) {
   9   CHECK_EQ(bottom.size(), 1) << "Neuron Layer takes a single blob as input.";
  10   CHECK_EQ(top->size(), 1) << "Neuron Layer takes a single blob as output.";
  11   (*top)[0]->Reshape(bottom[0]->num(), bottom[0]->channels(),
  12       bottom[0]->height(), bottom[0]->width());
  13 };
  14
  15 template class NeuronLayer<float>;
  16 template class NeuronLayer<double>;
  17
  18 }  // namespace caffeine