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