[ Property ] Add Tensor Type property in model
authorjijoong.moon <jijoong.moon@samsung.com>
Thu, 29 Jun 2023 12:36:30 +0000 (21:36 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Mon, 21 Aug 2023 06:29:23 +0000 (15:29 +0900)
This PR enables the tensor type in model property as
"tensor_type=NHWC" or "tensor_type=NCHW". This information goes to
network_grap and layer node & manager.

Then, each layer can get the model tensor type information and it can
be used to request tensor or just using temporal tensor.

Resolves:

**Self evaluation:**
1. Build test:  [X]Passed [ ]Failed [ ]Skipped
2. Run test:  [X]Passed [ ]Failed [ ]Skipped

Signed-off-by: jijoong.moon <jijoong.moon@samsung.com>
nntrainer/graph/network_graph.cpp

index 8142d91..a3f4af2 100644 (file)
@@ -81,6 +81,12 @@ int NetworkGraph::compile(const std::string &loss_type) {
 
   inPlaceOptimize();
 
+  TensorDim::Format type =
+    (getModelTensorType().compare("NCHW") ||
+     getModelTensorType().compare("nchw") || getModelTensorType().compare(""))
+      ? TensorDim::Format::NCHW
+      : TensorDim::Format::NHWC;
+
   for (auto iter = cbegin(); iter != cend(); iter++) {
     auto lnode = (*iter);
     /// @todo  later, we can set layer tensor type differenctly with model