From 30eac57ca182f784a178894d42628a62afbbb41c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/On-Device=20Lab=28SR=29/Princi?=
 =?UTF-8?q?pal=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?=
 <saehie.park@samsung.com>
Date: Tue, 17 Sep 2019 10:47:46 +0900
Subject: [exo-tflite] Resolve TensorBroadcast for Mul (#7491)

This will add handler for TFLMul node in TensorBroadcastConverter

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
---
 .../src/Conversion/TensorBroadcastConverter.cpp           | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/compiler/exo-tflite/src/Conversion/TensorBroadcastConverter.cpp b/compiler/exo-tflite/src/Conversion/TensorBroadcastConverter.cpp
index 1113c1189..84f74a2dd 100644
--- a/compiler/exo-tflite/src/Conversion/TensorBroadcastConverter.cpp
+++ b/compiler/exo-tflite/src/Conversion/TensorBroadcastConverter.cpp
@@ -52,7 +52,14 @@ struct Collector final : public locoex::TFLNodeMutableVisitor<void>
 
   // TODO ADD TFLDiv
 
-  // TODO ADD TFLMul
+  void visit(locoex::TFLMul *node) final
+  {
+    if (auto tbc = input_as_tbc<locoex::TFLMul>(node))
+    {
+      NodePair pair(tbc, node);
+      candidates.insert(pair);
+    }
+  }
 
   // TODO ADD TFLSub
 
@@ -127,7 +134,11 @@ bool TensorBroadcastConverter::run(loco::Graph *graph)
         changed = true;
       }
       // TODO ADD TFLDiv
-      // TODO ADD TFLMul
+      else if (auto tflmul = dynamic_cast<locoex::TFLMul *>(pair.second))
+      {
+        jump_connection<locoex::TFLMul>(tensorbroadcast, tflmul);
+        changed = true;
+      }
       // TODO ADD TFLSub
       else
       {
-- 
cgit v1.2.3