From: Parichay Kapoor <pk.kapoor@samsung.com>
Date: Fri, 12 Nov 2021 05:33:27 +0000 (+0900)
Subject: [layer] Add exportTo for concat layer
X-Git-Tag: accepted/tizen/unified/20220323.062643~193
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=27eeb1d3a73163d6c4565e708c24c1ec99d6e79d;p=platform%2Fcore%2Fml%2Fnntrainer.git

[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
--- 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
--- 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; }