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 (file)
--- 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);