1 files 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
       {