From: 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Date: Thu, 19 Sep 2019 01:31:42 +0000 (+0900)
Subject: [exo-tflite] Broadcast conversion for Div, Sub (#7591)
X-Git-Tag: submit/tizen/20191205.083104~1169
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=5dbd19840564d969ee3572f57488b82c0e2207db;p=platform%2Fcore%2Fml%2Fnnfw.git

[exo-tflite] Broadcast conversion for Div, Sub (#7591)

This will enable TensorBroadcastConverter to handle Div and Sub node

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
---

diff --git a/compiler/exo-tflite/src/Conversion/TensorBroadcastConverter.cpp b/compiler/exo-tflite/src/Conversion/TensorBroadcastConverter.cpp
index 84f74a2..b852131 100644
--- a/compiler/exo-tflite/src/Conversion/TensorBroadcastConverter.cpp
+++ b/compiler/exo-tflite/src/Conversion/TensorBroadcastConverter.cpp
@@ -50,7 +50,14 @@ struct Collector final : public locoex::TFLNodeMutableVisitor<void>
     }
   }
 
-  // TODO ADD TFLDiv
+  void visit(locoex::TFLDiv *node) final
+  {
+    if (auto tbc = input_as_tbc<locoex::TFLDiv>(node))
+    {
+      NodePair pair(tbc, node);
+      candidates.insert(pair);
+    }
+  }
 
   void visit(locoex::TFLMul *node) final
   {
@@ -61,7 +68,14 @@ struct Collector final : public locoex::TFLNodeMutableVisitor<void>
     }
   }
 
-  // TODO ADD TFLSub
+  void visit(locoex::TFLSub *node) final
+  {
+    if (auto tbc = input_as_tbc<locoex::TFLSub>(node))
+    {
+      NodePair pair(tbc, node);
+      candidates.insert(pair);
+    }
+  }
 
   void visit(locoex::TFLNode *) final { return; }
 
@@ -133,13 +147,21 @@ bool TensorBroadcastConverter::run(loco::Graph *graph)
         jump_connection<locoex::TFLAdd>(tensorbroadcast, tfladd);
         changed = true;
       }
-      // TODO ADD TFLDiv
+      else if (auto tfldiv = dynamic_cast<locoex::TFLDiv *>(pair.second))
+      {
+        jump_connection<locoex::TFLDiv>(tensorbroadcast, tfldiv);
+        changed = true;
+      }
       else if (auto tflmul = dynamic_cast<locoex::TFLMul *>(pair.second))
       {
         jump_connection<locoex::TFLMul>(tensorbroadcast, tflmul);
         changed = true;
       }
-      // TODO ADD TFLSub
+      else if (auto tflsub = dynamic_cast<locoex::TFLSub *>(pair.second))
+      {
+        jump_connection<locoex::TFLSub>(tensorbroadcast, tflsub);
+        changed = true;
+      }
       else
       {
         assert(false);