[Graph] change of inplace-type setting method

nntrainer/layers/flatten_layer.h

@@ -28,8 +28,9 @@ class FlattenLayer : public ReshapeLayer {
   /**
    * @brief     Constructor of Flatten Layer
    */
-  FlattenLayer() : ReshapeLayer(), flatten_props(
-    props::StartDimension(), props::EndDimension()) {}
+  FlattenLayer() :
+    ReshapeLayer(),
+    flatten_props(props::StartDimension(), props::EndDimension()) {}
 
   /**
    * @brief     Destructor of Flatten Layer

nntrainer/layers/identity_layer.h

@@ -21,6 +21,7 @@ namespace nntrainer {
 
 /**
  * @class   Identity Layer
+ * @brief   Identity Layer
  * @note    Identity layers takes multiple tensors as input, redirects to output
  * without doing nothing (or if unavoidable, copying)
  */
@@ -73,6 +74,17 @@ class IdentityLayer final : public Layer {
    */
   bool supportInPlace() const override { return true; }
 
+  /**
+   * @brief Initialize the in-place type of the layer
+   * @return InPlaceType
+   */
+  InPlaceType initializeInPlaceType() final {
+    if (!supportInPlace())
+      return InPlaceType::NONE;
+    else
+      return InPlaceType::RESTRICTING;
+  }
+
   /**
    * @copydoc Layer::getType()
    */

nntrainer/layers/layer_devel.h

@@ -41,6 +41,18 @@ class InitLayerContext;
 class RunLayerContext;
 class Exporter;
 
+/**
+ * @brief Enum class for the various types of inplace modes supported by layer
+ *
+ */
+enum class InPlaceType {
+  NONE,           /**< layer is not inplace */
+  RESTRICTING,    /**< layer is in-place and does place restriction on layers
+                    ahead of it to be in-place */
+  NON_RESTRICTING /**< layer is in-place and does NOT place restriction on the
+                    layers ahead of it to be in-place */
+};
+
 /**
  * @class   Layer Base class for layers
  * @brief   Base class for all layers
@@ -248,6 +260,22 @@ class Layer {
    */
   virtual bool supportInPlace() const { return false; }
 
+  /**
+   * @brief Initialize the in-place type of the layer
+   * @details If it is a layer that supports in-place, the default in-place type
+   * is NONE_RESTRICTING, but if there is a RESTRICTING type among the input
+   * layers, it is set to NONE in the network_graph.cpp.
+   * Layers with exceptional behavior such as No-Operation layers should
+   * override this function.
+   * @return InPlaceType
+   */
+  virtual InPlaceType initializeInPlaceType() {
+    if (!supportInPlace())
+      return InPlaceType::NONE;
+    else
+      return InPlaceType::NON_RESTRICTING;
+  }
+
   /**
    * @brief  check if this layer requires label to be passed
    * @note   if requireLabel() == true means, for now, that it is endpoint of a

nntrainer/layers/layer_node.cpp

@@ -930,6 +930,15 @@ bool LayerNode::supportInPlace() const {
   return layer->supportInPlace();
 }
 
+/**
+ * @brief Initialize the in-place type of the layer
+ * @return InPlaceType
+ */
+InPlaceType LayerNode::initializeInPlaceType() {
+  inplace_type = layer->initializeInPlaceType();
+  return inplace_type;
+}
+
 /**
  * @brief  check if this layer requires label to be passed
  */

nntrainer/layers/layer_node.h

@@ -55,18 +55,6 @@ class Packed;
 class LossScaleForMixed;
 } // namespace props
 
-/**
- * @brief Enum class for the various types of inplace modes supported by layer
- *
- */
-enum class InPlaceType {
-  NONE,           /**< layer is not inplace */
-  RESTRICTING,    /**< layer is in-place and does place restriction on layers
-                    ahead of it to be in-place */
-  NON_RESTRICTING /**< layer is in-place and does NOT place restriction on the
-                    layers ahead of it to be in-place */
-};
-
 /**
  * @class   LayerNode class
  * @brief   layer node class for the graph
@@ -365,6 +353,16 @@ class LayerNode final : public ml::train::Layer, public GraphNode {
    */
   bool supportInPlace() const;
 
+  /**
+   * @brief Initialize the in-place type of the layer
+   * @details If it is a layer that supports in-place, the default in-place type
+   * is NONE_RESTRICTING, but if there is a RESTRICTING type among the input
+   * layers, it is set to NONE in the network_graph.cpp.
+   * Layers with exceptional behavior such as No-Operation layers should
+   * override this function.
+   * @return InPlaceType
+   */
+  InPlaceType initializeInPlaceType();
   /**
    * @brief   Notify that this layer will execute in-place
    *

nntrainer/layers/multiout_layer.h

@@ -73,6 +73,17 @@ class MultiOutLayer : public Layer {
    */
   bool supportBackwarding() const override { return true; };
 
+  /**
+   * @brief Initialize the in-place type of the layer
+   * @return InPlaceType
+   */
+  InPlaceType initializeInPlaceType() final {
+    if (!supportInPlace())
+      return InPlaceType::NONE;
+    else
+      return InPlaceType::RESTRICTING;
+  }
+
   /**
    * @copydoc Layer::supportInPlace()
    */

nntrainer/layers/reshape_layer.h

@@ -78,6 +78,17 @@ class ReshapeLayer : public Layer {
    */
   bool supportInPlace() const override { return true; }
 
+  /**
+   * @brief Initialize the in-place type of the layer
+   * @return InPlaceType
+   */
+  InPlaceType initializeInPlaceType() final {
+    if (!supportInPlace())
+      return InPlaceType::NONE;
+    else
+      return InPlaceType::RESTRICTING;
+  }
+
   /**
    * @copydoc Layer::exportTo(Exporter &exporter, ml::train::ExportMethods
    * method)