From: 윤현식/On-Device Lab(SR)/Principal Engineer/삼성전자 <hyunsik.yoon@samsung.com>
Date: Mon, 2 Sep 2019 08:24:26 +0000 (+0900)
Subject: [exo-tflite] Adding TFLTypeInferenceRule into TypeInference (#7097)
X-Git-Tag: accepted/tizen/unified/20190903.052428~12
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=886c073a8a14ec4f6f7318b12c06ae5c487d6673;p=platform%2Fcore%2Fml%2Fnnfw.git

[exo-tflite] Adding TFLTypeInferenceRule into TypeInference (#7097)

This adds TFLTypeInferenceRule into TypeInference.

Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>
---

diff --git a/compiler/exo-tflite/src/TypeInference.cpp b/compiler/exo-tflite/src/TypeInference.cpp
index b6e7667..69a8a74 100644
--- a/compiler/exo-tflite/src/TypeInference.cpp
+++ b/compiler/exo-tflite/src/TypeInference.cpp
@@ -18,6 +18,9 @@
 
 #include "schema_generated.h"
 
+#include "Dialect/Service/TFLTypeInferenceRule.h"
+#include "Dialect/IR/TFLDialect.h"
+
 #include <loco/IR/CanonicalNode.h>
 #include <loco/IR/CanonicalNodeVisitor.h>
 #include <loco/IR/CanonicalDialect.h>
@@ -69,11 +72,13 @@ void TypeInference::run(loco::Graph *g)
 {
   loco::CanonicalTypeInferenceRule canonical_rule;
   locoex::COpTypeInferenceRule cop_rule; // rule for custom op
+  locoex::TFLTypeInferenceRule tfl_rule;
 
   loco::MultiDialectTypeInferenceRule rules;
 
   rules.bind(loco::CanonicalDialect::get(), &canonical_rule)
-      .bind(locoex::COpDialect::get(), &cop_rule);
+      .bind(locoex::COpDialect::get(), &cop_rule)
+      .bind(locoex::TFLDialect::get(), &tfl_rule);
 
   loco::apply(&rules).to(g);
 }