[layer] Add exportTo for concat layer

Concat layer has properties which were not being saved. This patch adds
the corresponding support.

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>

diff --git a/nntrainer/layers/concat_layer.cpp b/nntrainer/layers/concat_layer.cpp
index fdf964e7fe55504bae808155703710aeef039396..6c3662b33c93370872a59e8d1d6074e3ca20514c 100644 (file)
--- a/nntrainer/layers/concat_layer.cpp
+++ b/nntrainer/layers/concat_layer.cpp
@@ -194,4 +194,10 @@ void ConcatLayer::setProperty(const std::vector<std::string> &values) {
          std::to_string(values.size());
 }
 
+void ConcatLayer::exportTo(Exporter &exporter,
+                           const ExportMethods &method) const {
+  Layer::exportTo(exporter, method);
+  exporter.saveResult(concat_props, method, this);
+}
+
 } /* namespace nntrainer */

diff --git a/nntrainer/layers/concat_layer.h b/nntrainer/layers/concat_layer.h
index 177e44ec520bd40657647ec20dcf49736cc51a31..11764e44f9b9e3fd28f917c2077bf22b4bfaf8f6 100644 (file)
--- a/nntrainer/layers/concat_layer.h
+++ b/nntrainer/layers/concat_layer.h
@@ -67,6 +67,11 @@ public:
    */
   const std::string getType() const override { return ConcatLayer::type; };
 
+  /**
+   * @copydoc Layer::exportTo(Exporter &exporter, ExportMethods method)
+   */
+  void exportTo(Exporter &exporter, const ExportMethods &method) const override;
+
   /**
    * @copydoc Layer::supportBackwarding()
    */