add ceil & floor layer

Author: vera121

FEATURES.md

@@ -9,11 +9,13 @@ evconvert (TensorFlow/ONNX/... to Caffe Converter) related
 atan_layer  
 batch_to_space_nd_layer  
 broadcast_to_layer  
-crop_and_resize  
+crop_and_resize_layer  
+ceil_layer  
 depth_to_space_layer  
 div_layer  
 expand_dims_nd_layer  
 floor_div_layer  
+floor_layer  
 floor_mod_layer  
 gather_layer  
 gather_nd_layer  

include/caffe/layers/ceil_layer.hpp

@@ -0,0 +1,37 @@
+#ifndef CAFFE_CEIL_LAYER_HPP_
+#define CAFFE_CEIL_LAYER_HPP_
+
+#include <vector>
+
+#include "caffe/blob.hpp"
+#include "caffe/layer.hpp"
+#include "caffe/proto/caffe.pb.h"
+
+namespace caffe {
+
+template <typename Dtype>
+class CeilLayer : public Layer<Dtype> {
+public:
+	explicit CeilLayer(const LayerParameter& param) :
+			Layer<Dtype>(param) {}
+	virtual void Reshape(const vector<Blob<Dtype>*>& bottom,
+			const vector<Blob<Dtype>*>& top);
+
+	virtual inline const char* type() const {return "Ceil";}
+	virtual inline int ExactBottomBlobs() const {return 1;}
+	virtual inline int ExactNumTopBlobs() const {return 1;}
+
+protected:
+	virtual void Forward_cpu(const vector<Blob<Dtype>*>& bottom,
+			const vector<Blob<Dtype>*>& top);
+	virtual void Backward_cpu(const vector<Blob<Dtype>*>& top,
+			const vector<bool>& propagate_down,
+			const vector<Blob<Dtype>*>& bottom) {
+		NOT_IMPLEMENTED;
+	}
+
+};
+
+}  // namespace caffe
+
+#endif  // CAFFE_CEIL_LAYER_HPP_

include/caffe/layers/floor_layer.hpp

@@ -0,0 +1,37 @@
+#ifndef CAFFE_FLOOR_LAYER_HPP_
+#define CAFFE_FLOOR_LAYER_HPP_
+
+#include <vector>
+
+#include "caffe/blob.hpp"
+#include "caffe/layer.hpp"
+#include "caffe/proto/caffe.pb.h"
+
+namespace caffe {
+
+template <typename Dtype>
+class FloorLayer : public Layer<Dtype> {
+public:
+	explicit FloorLayer(const LayerParameter& param) :
+			Layer<Dtype>(param) {}
+	virtual void Reshape(const vector<Blob<Dtype>*>& bottom,
+			const vector<Blob<Dtype>*>& top);
+
+	virtual inline const char* type() const {return "Floor";}
+	virtual inline int ExactBottomBlobs() const {return 1;}
+	virtual inline int ExactNumTopBlobs() const {return 1;}
+
+protected:
+	virtual void Forward_cpu(const vector<Blob<Dtype>*>& bottom,
+			const vector<Blob<Dtype>*>& top);
+	virtual void Backward_cpu(const vector<Blob<Dtype>*>& top,
+			const vector<bool>& propagate_down,
+			const vector<Blob<Dtype>*>& bottom) {
+		NOT_IMPLEMENTED;
+	}
+
+};
+
+}  // namespace caffe
+
+#endif  // CAFFE_FLOOR_LAYER_HPP_

src/caffe/layers/ceil_layer.cpp

@@ -0,0 +1,33 @@
+#include <algorithm>
+#include <functional>
+#include <utility>
+#include <vector>
+
+#include "caffe/layers/ceil_layer.hpp"
+
+namespace caffe {
+template <typename Dtype>
+void CeilLayer<Dtype>::Reshape(const vector<Blob<Dtype>*>& bottom,
+      const vector<Blob<Dtype>*>& top) {
+  const int count = bottom[0]->count();
+  //CHECK_NE(top[0], bottom[0]) << this->type() << " Layer does not "
+  //      "allow in-place computation.";
+  top[0]->ReshapeLike(*bottom[0]);
+  CHECK_EQ(count, top[0]->count());
+}
+
+template <typename Dtype>
+void CeilLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom,
+    const vector<Blob<Dtype>*>& top) {
+	  const Dtype* bottom_data = bottom[0]->cpu_data();
+	  Dtype* top_data = top[0]->mutable_cpu_data();
+	  const int count = bottom[0]->count();
+	  for (int i = 0; i < count; ++i) {
+	    top_data[i] = ceil(bottom_data[i]);
+	  }
+}
+
+INSTANTIATE_CLASS(CeilLayer);
+REGISTER_LAYER_CLASS(Ceil);
+
+}  // namespace caffe

src/caffe/layers/floor_layer.cpp

@@ -0,0 +1,33 @@
+#include <algorithm>
+#include <functional>
+#include <utility>
+#include <vector>
+
+#include "caffe/layers/floor_layer.hpp"
+
+namespace caffe {
+template <typename Dtype>
+void FloorLayer<Dtype>::Reshape(const vector<Blob<Dtype>*>& bottom,
+      const vector<Blob<Dtype>*>& top) {
+  const int count = bottom[0]->count();
+  //CHECK_NE(top[0], bottom[0]) << this->type() << " Layer does not "
+  //      "allow in-place computation.";
+  top[0]->ReshapeLike(*bottom[0]);
+  CHECK_EQ(count, top[0]->count());
+}
+
+template <typename Dtype>
+void FloorLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom,
+    const vector<Blob<Dtype>*>& top) {
+	  const Dtype* bottom_data = bottom[0]->cpu_data();
+	  Dtype* top_data = top[0]->mutable_cpu_data();
+	  const int count = bottom[0]->count();
+	  for (int i = 0; i < count; ++i) {
+	    top_data[i] = floor(bottom_data[i]);
+	  }
+}
+
+INSTANTIATE_CLASS(FloorLayer);
+REGISTER_LAYER_CLASS(Floor);
+
+}  // namespace caffe