Update sigmoid output from Y to out, test=develop (#24765)

5 years ago · dbcd7c69e9
parent b0e7439fbc
commit dbcd7c69e9
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
+++ b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
@ -83,7 +83,7 @@ _op_real_in_out_name = {
    "swish": [["X"], ["Out"]],
    "dropout": [["X"], ["Out"]],
    "batch_norm": [["X"], ["Y"]],
-    "sigmoid": [["X"], ["Y"]],
+    "sigmoid": [["X"], ["Out"]],
 }