fix relu6 bug in QuantizationOptimizer.hpp

huawei-noah · Aug 25, 2024 · fafe6c8 · fafe6c8
1 parent 31044be
commit fafe6c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)