temporary code for layer initialization

 - Temporary code for layer initialization

Signed-off-by: hyeonseok lee <hs89.lee@samsung.com>

diff --git a/api/ccapi/include/layer.h b/api/ccapi/include/layer.h
index 19266ae5a760f15ee8a77941f046876b29771571..8bd13e01bc8e867053482bd0995563985b6817b5 100644 (file)
--- a/api/ccapi/include/layer.h
+++ b/api/ccapi/include/layer.h
@@ -131,6 +131,8 @@ public:
    */
   virtual const std::string getType() const = 0;
 
+  virtual void initialize() = 0;
+
   /**
    * @brief     Default allowed properties
    * - input shape : string

diff --git a/nntrainer/layers/layer_devel.h b/nntrainer/layers/layer_devel.h
index 54ce1a0ee903c62366ee35b42eec2037d7d35bb5..be02b5bc2fac91eb1ff1e0c58cea3258ca7fcaf8 100644 (file)
--- a/nntrainer/layers/layer_devel.h
+++ b/nntrainer/layers/layer_devel.h
@@ -160,6 +160,8 @@ public:
    */
   virtual void finalize(InitLayerContext &context) = 0;
 
+  virtual void initialize(RunLayerContext &context){};
+
   /**
    * @brief     Forward Propagation of a layer
    * @param     context Context of the layer

diff --git a/nntrainer/layers/layer_node.h b/nntrainer/layers/layer_node.h
index 55b24f44e9656d923956e41ccd58583d45e346bf..ccd8bb8be10c8cfab225e328c90b6bdac5b24d1b 100644 (file)
--- a/nntrainer/layers/layer_node.h
+++ b/nntrainer/layers/layer_node.h
@@ -273,6 +273,8 @@ public:
    */
   InitLayerContext refinalize(const std::vector<TensorDim> &input_dims = {});
 
+  void initialize() override { layer->initialize(*run_context); }
+
   /**
    * @brief     Forward Propagation of a layer
    * @param     training true if training, false if inference