[luci] Quantize BroadcastTo Op activation

This quantizes activation for BroadcastTo. ONE-DCO-1.0-Signed-off-by: Hyukjin Jeong <[email protected]>
Samsung · Dec 16, 2024 · c4da132 · c4da132
1 parent b5a9598
commit c4da132
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/compiler/luci/pass/src/QuantizeActivation.cpp b/compiler/luci/pass/src/QuantizeActivation.cpp
@@ -222,6 +222,7 @@ void QuantizeConstInputActivation::visit(luci::CircleNode *node)
 QUANTIZE_SINGLE_CONST_INPUT(luci::CircleAbs, x)
 QUANTIZE_SINGLE_CONST_INPUT(luci::CircleArgMax, input)
 QUANTIZE_SINGLE_CONST_INPUT(luci::CircleArgMin, input)
+QUANTIZE_SINGLE_CONST_INPUT(luci::CircleBroadcastTo, input)
 QUANTIZE_SINGLE_CONST_INPUT(luci::CircleBatchToSpaceND, input)
 QUANTIZE_SINGLE_CONST_INPUT(luci::CircleDepthToSpace, input)
 QUANTIZE_SINGLE_CONST_INPUT(luci::CircleElu, features)

diff --git a/compiler/luci/pass/src/QuantizeActivation.h b/compiler/luci/pass/src/QuantizeActivation.h
@@ -98,6 +98,7 @@ struct QuantizeConstInputActivation final : public luci::CircleNodeMutableVisito
   void visit(luci::CircleAbs *node);
   void visit(luci::CircleArgMax *node);
   void visit(luci::CircleArgMin *node);
+  void visit(luci::CircleBroadcastTo *node);
   void visit(luci::CircleBatchToSpaceND *node);
   void visit(luci::CircleDepthToSpace *node);
   void visit(luci::CircleElu *node);