code style changes

leeburt · Jan 28, 2018 · f133729 · f133729
1 parent 03621aa
commit f133729
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 8 deletions.
diff --git a/src/layer/shufflechannel.cpp b/src/layer/shufflechannel.cpp
@@ -18,6 +18,19 @@ namespace ncnn {
 
 DEFINE_LAYER_CREATOR(ShuffleChannel)
 
+ShuffleChannel::ShuffleChannel()
+{
+    one_blob_only = true;
+    support_inplace = false;
+}
+
+int ShuffleChannel::load_param(const ParamDict& pd)
+{
+    group = pd.get(0, 1);
+
+    return 0;
+}
+
 int ShuffleChannel::forward(const Mat &bottom_blob, Mat &top_blob) const
 {
     int c = bottom_blob.c;
@@ -31,6 +44,7 @@ int ShuffleChannel::forward(const Mat &bottom_blob, Mat &top_blob) const
         // reject invalid group
         return -100;
     }
+
     top_blob.create(w, h, c);
     if (top_blob.empty())
         return -100;

diff --git a/src/layer/shufflechannel.h b/src/layer/shufflechannel.h
@@ -22,14 +22,10 @@ namespace ncnn {
 class ShuffleChannel : public Layer
 {
 public:
-    ShuffleChannel() {
-        one_blob_only = true;
-        support_inplace = false;
-    }
-    virtual int load_param(const ParamDict& pd) {
-        group = pd.get(0, 1);
-        return 0;
-    }
+    ShuffleChannel();
+
+    virtual int load_param(const ParamDict& pd);
+
     virtual int forward(const Mat& bottom_blob, Mat& top_blob) const;
 
 public: