[exo-tflite] Enabling OperationExporter to handle TFLNode (#7001)

This enables OperationExporter to handle TFLNode

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

diff --git a/compiler/exo-tflite/src/OperationExporter.cpp b/compiler/exo-tflite/src/OperationExporter.cpp
index d5b0c8a..f0108ec 100644 (file)
--- a/compiler/exo-tflite/src/OperationExporter.cpp
+++ b/compiler/exo-tflite/src/OperationExporter.cpp
@@ -18,6 +18,7 @@
 #include "ExporterUtils.h"
 #include "ShapeInference.h"
 
+#include "Dialect/IR/TFLNode.h"
 #include "Dialect/IR/TFLNodes.h"
 #include "Dialect/IR/TFLNodeVisitor.h"
 
@@ -628,10 +629,15 @@ void exportNode(loco::Node *node, flatbuffers::FlatBufferBuilder &builder,
   }
 
   if (auto canonical_node = dynamic_cast<loco::CanonicalNode *>(node))
-  {
+  { // TODO Consider removing this later
     OperationExporter exporter{builder, data};
     canonical_node->accept(&exporter);
   }
+  else if (auto tfl_node = dynamic_cast<locoex::TFLNode *>(node))
+  {
+    OperationExporter exporter{builder, data};
+    tfl_node->accept(&exporter);
+  }
   else if (dynamic_cast<locoex::COpNode *>(node))
   {
     OperationExporter exporter{builder, data};