[exo-tflite] Adding TFLShapeInferenceRule into ShapeInference (#7096)

* [exo-tflite] Adding TFLShapeInferenceRule into ShapeInference

This adds TFLShapeInferenceRule into ShapeInference.

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

diff --git a/compiler/exo-tflite/src/ShapeInference.cpp b/compiler/exo-tflite/src/ShapeInference.cpp
index d71fed9..4ca4599 100644 (file)
--- a/compiler/exo-tflite/src/ShapeInference.cpp
+++ b/compiler/exo-tflite/src/ShapeInference.cpp
@@ -15,6 +15,8 @@
  */
 
 #include "ShapeInference.h"
+#include "Dialect/IR/TFLDialect.h"
+#include "Dialect/Service/TFLShapeInferenceRule.h"
 
 #include <loco/IR/CanonicalNode.h>
 #include <loco/IR/CanonicalDialect.h>
@@ -32,10 +34,13 @@ void ShapeInference::run(loco::Graph *g)
   // TODO Adjust indentation level
   {
     loco::CanonicalShapeInferenceRule canonical_rule;
+    locoex::TFLShapeInferenceRule tfl_rule;
     locoex::COpShapeInferenceRule cop_rule;
+
     loco::MultiDialectShapeInferenceRule rules;
 
     rules.bind(loco::CanonicalDialect::get(), &canonical_rule)
+        .bind(locoex::TFLDialect::get(), &tfl_rule)
         .bind(locoex::COpDialect::get(), &cop_rule);
 
     loco::apply(&rules).to(g);