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);