From cc4ad575d668db1d3fbe27d492b4647a4b8444ae Mon Sep 17 00:00:00 2001
From: "Efimov Alexander/AI Tools Lab/./Samsung Electronics"
 <a.efimov@samsung.com>
Date: Mon, 16 Jul 2018 20:17:25 +0300
Subject: [PATCH] Inference shape in codegen (#674)

Add shape inference in codegen

Signed-off-by: Efimov Alexander <a.efimov@samsung.com>
---
 contrib/nnc/libs/backend/soft/src/generator.cpp | 4 ++++
 1 file changed, 4 insertions(+)

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);
-- 
2.7.4