From c4da1320aefd665694b2d21ad15672c53cde78bf Mon Sep 17 00:00:00 2001 From: Hyukjin Jeong Date: Mon, 16 Dec 2024 18:45:08 +0900 Subject: [PATCH] [luci] Quantize BroadcastTo Op activation This quantizes activation for BroadcastTo. ONE-DCO-1.0-Signed-off-by: Hyukjin Jeong --- compiler/luci/pass/src/QuantizeActivation.cpp | 1 + compiler/luci/pass/src/QuantizeActivation.h | 1 + 2 files changed, 2 insertions(+) diff --git a/compiler/luci/pass/src/QuantizeActivation.cpp b/compiler/luci/pass/src/QuantizeActivation.cpp index f32c47fdc5f..b15b67a935a 100644 --- 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 index c9a02b39801..f49916e1452 100644 --- 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);