apache · ghost · Sep 17, 2015
@@ -8,6 +8,7 @@
 #define MXNET_OPERATOR_H_
 
 #include <dmlc/base.h>
+#include <dmlc/json.h>
 #include <dmlc/logging.h>
 #include <dmlc/registry.h>
 #include <vector>
@@ -389,6 +390,9 @@ class OperatorProperty {
    * \return a new constructed OperatorProperty
    */
   static OperatorProperty *Create(const char* type_name);
+
+  virtual void Save(dmlc::JSONWriter *writer) const = 0;
+  virtual void Load(dmlc::JSONReader *reader) = 0;
 };
 
 /*! \brief typedef the factory function of operator property */

@@ -8,6 +8,8 @@
 #define MXNET_SYMBOLIC_H_
 
 #include <dmlc/base.h>
+#include <dmlc/json.h>
+#include <algorithm>
 #include <vector>
 #include <memory>
 #include <string>
@@ -64,6 +66,11 @@ class StaticGraph {
       if (source_id == other.source_id) return index < other.index;
       return source_id < other.source_id;
     }
+
+    /*! \brief interface for json serialization */
+    void Save(dmlc::JSONWriter *writer) const;
+    /*! \brief interface for json serialization */
+    void Load(dmlc::JSONReader *reader);
   };
   /*!
    * \brief Operation Node in static graphs.
@@ -95,6 +102,23 @@ class StaticGraph {
     int32_t backward_source_id;
     /*! \brief default constructor */
     Node() : backward_source_id(-1) {}
+
+    friend void swap(Node& lhs, Node& rhs) {
+      std::swap(lhs.op, rhs.op);
+      std::swap(lhs.name, rhs.name);
+      std::swap(lhs.inputs, rhs.inputs);
+      std::swap(lhs.backward_source_id, rhs.backward_source_id);
+    }
+    /*! \brief copy constructor in favor of serialization. */
+    Node(const Node& another) : op(another.op.get() ? another.op.get()->Copy() : nullptr),
+                                name(another.name),
+                                inputs(another.inputs),
+                                backward_source_id(another.backward_source_id) {}
+
+    inline Node& operator=(Node another) {
+      swap(*this, another);
+      return *this;
+    }
     /*! \return whether the node is forward op node */
     inline bool is_forward() const {
       return op != nullptr;
@@ -107,13 +131,25 @@ class StaticGraph {
     inline bool is_variable() const {
       return op == nullptr && !is_backward();
     }
+    /*! \brief interface for json serialization */
+    void Save(dmlc::JSONWriter *writer) const;
+    /*! \brief interface for json serialization */
+    void Load(dmlc::JSONReader *reader);
   };
   /*! \brief all nodes in the graph */
   std::vector<Node> nodes;
   /*! \brief index of nodes that correspods to arguments */
   std::vector<uint32_t> arg_nodes;
   /*! \brief heads outputs of the graph */
   std::vector<DataEntry> heads;
+  /*! \brief load static graph from json. TODO: a static creator's better */
+  void Load(const std::string& json);
+  /*! \brief save static graph to json */
+  void Save(std::string* json) const;
+  /*! \brief interface for json serialization */
+  void Save(dmlc::JSONWriter *writer) const;
+  /*! \brief interface for json serialization */
+  void Load(dmlc::JSONReader *reader);
   // funtions to help inference in static graph
   /*!
    * \brief Perform a topological sort on the graph

diff --git a/src/operator/activation-inl.h b/src/operator/activation-inl.h
@@ -84,7 +84,7 @@ template<typename xpu>
 Operator* CreateOp(ActivationParam type);
 
 #if DMLC_USE_CXX11
-class ActivationProp : public OperatorProperty {
+class ActivationProp : public ParamOperatorProperty<ActivationParam> {
  public:
   void Init(const std::vector<std::pair<std::string, std::string> >& kwargs) override {
     param_.Init(kwargs);
@@ -139,12 +139,8 @@ class ActivationProp : public OperatorProperty {
   }
 
   Operator* CreateOperator(Context ctx) const;
-
- private:
-  ActivationParam param_;
 };
 #endif  // DMLC_USE_CXX11
 }  // namespace op
 }  // namespace mxnet
 #endif  // MXNET_OPERATOR_ACTIVATION_INL_H_
-
diff --git a/src/operator/batch_norm-inl.h b/src/operator/batch_norm-inl.h
@@ -182,7 +182,7 @@ Operator *CreateOp(BatchNormParam param);
 
 
 #if DMLC_USE_CXX11
-class BatchNormProp : public OperatorProperty {
+class BatchNormProp : public ParamOperatorProperty<BatchNormParam> {
  public:
   void Init(const std::vector<std::pair<std::string, std::string> >& kwargs) override {
     param_.Init(kwargs);
@@ -263,8 +263,6 @@ class BatchNormProp : public OperatorProperty {
 
   Operator* CreateOperator(Context ctx) const;
 
- private:
-  BatchNormParam param_;
 };  // class BatchNormProp
 
 #endif  // DMLC_USE_CXX11

diff --git a/src/operator/concat-inl.h b/src/operator/concat-inl.h
@@ -163,7 +163,7 @@ template<typename xpu>
 Operator *CreateOp(ConcatParam param);
 
 #if DMLC_USE_CXX11
-class ConcatProp : public OperatorProperty {
+class ConcatProp : public ParamOperatorProperty<ConcatParam> {
  public:
   void Init(const std::vector<std::pair<std::string, std::string> >& kwargs) override {
     param_.Init(kwargs);
@@ -223,9 +223,6 @@ class ConcatProp : public OperatorProperty {
   }
 
   Operator* CreateOperator(Context ctx) const;
-
- private:
-  ConcatParam param_;
 };  // class ConcatProp
 #endif  // DMLC_USE_CXX11
 }  // namespace op

diff --git a/src/operator/convolution-inl.h b/src/operator/convolution-inl.h
@@ -262,7 +262,7 @@ template<typename xpu>
 Operator* CreateOp(ConvolutionParam param);
 
 #if DMLC_USE_CXX11
-class ConvolutionProp : public OperatorProperty {
+class ConvolutionProp : public ParamOperatorProperty<ConvolutionParam> {
  public:
   std::vector<std::string> ListArguments() const override {
     if (!param_.no_bias) {
@@ -358,8 +358,6 @@ class ConvolutionProp : public OperatorProperty {
 
   Operator* CreateOperator(Context ctx) const;
 
- private:
-  ConvolutionParam param_;
 };  // class ConvolutionProp
 #endif  // DMLC_USE_CXX11
 }  // namespace op

diff --git a/src/operator/dropout-inl.h b/src/operator/dropout-inl.h
@@ -91,7 +91,7 @@ template<typename xpu>
 Operator *CreateOp(DropoutParam param);
 
 #if DMLC_USE_CXX11
-class DropoutProp : public OperatorProperty {
+class DropoutProp : public ParamOperatorProperty<DropoutParam> {
  public:
   void Init(const std::vector<std::pair<std::string, std::string> >& kwargs) override {
     param_.Init(kwargs);
@@ -160,8 +160,6 @@ class DropoutProp : public OperatorProperty {
 
   Operator* CreateOperator(Context ctx) const;
 
- private:
-  DropoutParam param_;
 };  // class DropoutProp
 #endif  // DMLC_USE_CXX11
 }  // namespace op

diff --git a/src/operator/elementwise_binary_op-inl.h b/src/operator/elementwise_binary_op-inl.h
@@ -157,7 +157,7 @@ Operator* CreateElementWiseBinaryOp(ElementWiseBinaryOpType type);
 
 #if DMLC_USE_CXX11
 template<typename ForwardOp>
-class ElementWiseBinaryOpProp : public OperatorProperty {
+class ElementWiseBinaryOpProp : public NoParamOperatorProperty {
  public:
   void Init(const std::vector<std::pair<std::string, std::string> >& kwargs) override {
     CHECK_EQ(kwargs.size(), 0)

diff --git a/src/operator/elementwise_sum-inl.h b/src/operator/elementwise_sum-inl.h
@@ -102,6 +102,16 @@ class ElementWiseSumOp : public Operator {
       Assign(igrad, req[i], F<mshadow_op::identity>(ograd));
     }
   }
+  inline void Save(dmlc::JSONWriter *writer) const {
+    writer->BeginObject();
+    writer->WriteObjectKeyValue("size_", size_);
+    writer->EndObject();
+  }
+  inline void Load(dmlc::JSONReader *reader) {
+    dmlc::JSONObjectReadHelper helper;
+    helper.DeclareField("size_", &size_);
+    helper.ReadAllFields(reader);
+  }
 
  private:
   int size_;
@@ -111,7 +121,7 @@ template<typename xpu>
 Operator* CreateOp(ElementWiseSumParam param);
 
 #if DMLC_USE_CXX11
-class ElementWiseSumProp : public OperatorProperty {
+class ElementWiseSumProp : public ParamOperatorProperty<ElementWiseSumParam> {
  public:
   void Init(const std::vector<std::pair<std::string, std::string> >& kwargs) override {
     param_.Init(kwargs);
@@ -180,9 +190,6 @@ class ElementWiseSumProp : public OperatorProperty {
   }
 
   Operator* CreateOperator(Context ctx) const;
-
- private:
-  ElementWiseSumParam param_;
 };  // class ElementWiseSumProp
 
 #endif  // DMLC_USE_CXX11

diff --git a/src/operator/fully_connected-inl.h b/src/operator/fully_connected-inl.h
@@ -124,7 +124,7 @@ template<typename xpu>
 Operator* CreateOp(FullyConnectedParam param);
 
 #if DMLC_USE_CXX11
-class FullyConnectedProp : public OperatorProperty {
+class FullyConnectedProp : public ParamOperatorProperty<FullyConnectedParam> {
  public:
   std::vector<std::string> ListArguments() const override {
     if (!param_.no_bias) {
@@ -189,9 +189,6 @@ class FullyConnectedProp : public OperatorProperty {
   }
 
   Operator* CreateOperator(Context ctx) const;
-
- private:
-  FullyConnectedParam param_;
 };  // class FullyConnectedSymbol
 #endif
 }  // namespace op

diff --git a/src/operator/leaky_relu-inl.h b/src/operator/leaky_relu-inl.h
@@ -190,7 +190,7 @@ template<typename xpu>
 Operator* CreateOp(LeakyReLUParam type);
 
 #if DMLC_USE_CXX11
-class LeakyReLUProp : public OperatorProperty {
+class LeakyReLUProp : public ParamOperatorProperty<LeakyReLUParam> {
  public:
   void Init(const std::vector<std::pair<std::string, std::string> >& kwargs) override {
     param_.Init(kwargs);
@@ -298,9 +298,6 @@ class LeakyReLUProp : public OperatorProperty {
   }
 
   Operator* CreateOperator(Context ctx) const;
-
- private:
-  LeakyReLUParam param_;
 };
 #endif  // DMLC_USE_CXX11
 }  // namespace op

diff --git a/src/operator/lrn-inl.h b/src/operator/lrn-inl.h
@@ -98,7 +98,7 @@ template<typename xpu>
 Operator *CreateOp(LRNParam param);
 
 #if DMLC_USE_CXX11
-class LocalResponseNormProp : public OperatorProperty {
+class LocalResponseNormProp : public ParamOperatorProperty<LRNParam> {
  public:
   void Init(const std::vector<std::pair<std::string, std::string> >& kwargs) override {
     param_.Init(kwargs);
@@ -173,9 +173,6 @@ class LocalResponseNormProp : public OperatorProperty {
   }
 
   Operator* CreateOperator(Context ctx) const;
-
- private:
-  LRNParam param_;
 };  // LocalResponseNormProp
 #endif  // DMLC_USE_CXX11
 }  // namespace op

diff --git a/src/operator/operator_common.h b/src/operator/operator_common.h
@@ -8,9 +8,12 @@
 #ifndef MXNET_OPERATOR_OPERATOR_COMMON_H_
 #define MXNET_OPERATOR_OPERATOR_COMMON_H_
 
+#include <dmlc/json.h>
 #include <dmlc/logging.h>
 #include <mxnet/operator.h>
 #include <mxnet/base.h>
+#include <istream>
+#include <ostream>
 #include <string>
 
 namespace mxnet {
@@ -93,6 +96,43 @@ struct InferShapeError {
   }
 #endif
 
+#if DMLC_USE_CXX11
+template<class Param>
+class ParamOperatorProperty : public OperatorProperty {
+ public:
+  ParamOperatorProperty() {}
+  explicit ParamOperatorProperty(Param param) : param_(param) {}
+  inline void Save(dmlc::JSONWriter *writer) const {
+    writer->BeginObject();
+    std::string value = param_.PrintJson();
+    writer->WriteObjectKeyValue("param", value);
+    writer->EndObject();
+  }
+  inline void Load(dmlc::JSONReader *reader) {
+    dmlc::JSONObjectReadHelper helper;
+    std::string value;
+    helper.DeclareField("param", &value);
+    helper.ReadAllFields(reader);
+    param_.LoadJson(value);
+  }
+  inline bool operator==(const ParamOperatorProperty<Param>& other) const {
+    return param_ == other.param_;
+  }
+ protected:
+  Param param_;
+};
+
+class NoParamOperatorProperty : public OperatorProperty {
+ public:
+  inline void Save(dmlc::JSONWriter *writer) const {
+  }
+  inline void Load(dmlc::JSONReader *reader) {
+  }
+  inline bool operator==(const NoParamOperatorProperty& other) const {
+    return true;
+  }
+};
+#endif  // DMLC_USE_CXX11
 }  // namespace op
 }  // namespace mxnet
 #endif  // MXNET_OPERATOR_OPERATOR_COMMON_H_
diff --git a/src/operator/param.h b/src/operator/param.h
@@ -71,4 +71,3 @@ struct Param {
 }  // namespace mxnet
 
 #endif  // MXNET_OPERATOR_PARAM_H_
-
diff --git a/src/operator/pooling-inl.h b/src/operator/pooling-inl.h
@@ -154,7 +154,7 @@ Operator* CreateOp(PoolingParam param);
 
 
 #if DMLC_USE_CXX11
-class PoolingProp : public OperatorProperty {
+class PoolingProp : public ParamOperatorProperty<PoolingParam> {
  public:
   void Init(const std::vector<std::pair<std::string, std::string> >& kwargs) override {
     param_.Init(kwargs);
@@ -209,9 +209,6 @@ class PoolingProp : public OperatorProperty {
   }
 
   Operator* CreateOperator(Context ctx) const;
-
- private:
-  PoolingParam param_;
 };  // class PoolingProp
 #endif  // DMLC_USE_CXX11
 }  // namespace op

diff --git a/src/operator/reshape-inl.h b/src/operator/reshape-inl.h
@@ -83,11 +83,11 @@ template<typename xpu>
 Operator* CreateOp();
 
 #if DMLC_USE_CXX11
-class ReshapeProp : public OperatorProperty {
+class ReshapeProp : public ParamOperatorProperty<ReshapeParam> {
  public:
   ReshapeProp() {}
 
-  explicit ReshapeProp(ReshapeParam param) : param_(param) {}
+  explicit ReshapeProp(ReshapeParam param) : ParamOperatorProperty<ReshapeParam>(param) {}
 
   void Init(const std::vector<std::pair<std::string, std::string> >& kwargs) override {
     param_.Init(kwargs);
@@ -140,9 +140,6 @@ class ReshapeProp : public OperatorProperty {
   }
 
   Operator* CreateOperator(Context ctx) const;
-
- private:
-  ReshapeParam param_;
 };  // class ReshapeProp
 
 class FlattenProp : public ReshapeProp {
Original file line number	Diff line number	Diff line change
Expand Up		@@ -71,4 +71,3 @@ struct Param {
		} // namespace mxnet

		#endif // MXNET_OPERATOR_PARAM_H_