From: Efimov Alexander/AI Tools Lab/./Samsung Electronics <a.efimov@samsung.com>
Date: Mon, 16 Jul 2018 17:17:25 +0000 (+0300)
Subject: Inference shape in codegen (#674)
X-Git-Tag: nncc_backup~2410
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=cc4ad575d668db1d3fbe27d492b4647a4b8444ae;p=platform%2Fcore%2Fml%2Fnnfw.git

Inference shape in codegen (#674)

Add shape inference in codegen

Signed-off-by: Efimov Alexander <a.efimov@samsung.com>
---

diff --git a/contrib/nnc/libs/backend/soft/src/generator.cpp b/contrib/nnc/libs/backend/soft/src/generator.cpp
index aa4a4f3..9254051 100644
--- a/contrib/nnc/libs/backend/soft/src/generator.cpp
+++ b/contrib/nnc/libs/backend/soft/src/generator.cpp
@@ -1,6 +1,7 @@
 #include "generator.h"
 #include "model_analyzer.h"
 #include "PluginException.h"
+#include "nnc/core/IR/model/actions/ShapeInference.h"
 
 #include <sys/types.h>
 #include <sys/stat.h>
@@ -144,6 +145,9 @@ void BaseCodeGenerator::materializeModelParams(ostream &out, const ModelAnalyzer
 
 void BaseCodeGenerator::generate(Graph *g)
 {
+  // inference shapes
+  core::IR::model::ShapeInference si;
+  g->accept(&si);
   // visit and analyze graph
   ModelAnalyzer ma;
   g->accept(&ma);