[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 fdf964e..6c3662b 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 177e44e..11764e4 100644 (file)
--- a/nntrainer/layers/concat_layer.h
+++ b/nntrainer/layers/concat_layer.h
@@ -68,6 +68,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()
    */
   bool supportBackwarding() const override { return true; }