diff --git a/model_tools/include/OPOptimizers/QuantizationOptimizer.hpp b/model_tools/include/OPOptimizers/QuantizationOptimizer.hpp index c032a210..8fbf728d 100644 --- a/model_tools/include/OPOptimizers/QuantizationOptimizer.hpp +++ b/model_tools/include/OPOptimizers/QuantizationOptimizer.hpp @@ -219,7 +219,7 @@ class QuantizationOptimizer : public OPOptimizer { bool isNotNaiveRelu(OperatorSpec *op) { - return ((op->type == OT_Relu) && (op->ps.relu_spec.neg_slope != 0)); + return ((op->type == OT_Relu && op->ps.relu_spec.neg_slope != 0) || op->type == OT_Relu6); } bool isAvgPooling(OperatorSpec *op)