Merge pull request #5 from neuzxy/quant

add quant feature in fused_seqpool_cvm_op
jack603047588 · Sep 27, 2020 · ae11126 · ae11126
2 parents 5fac7a4 + 1c5c059
commit ae11126
Show file tree

Hide file tree

Showing 3 changed files with 152 additions and 156 deletions.
diff --git a/paddle/fluid/operators/fused/fused_seqpool_cvm_op.cc b/paddle/fluid/operators/fused/fused_seqpool_cvm_op.cc
@@ -107,6 +107,10 @@ class FusedSeqpoolCVMOpMaker : public framework::OpProtoAndCheckerMaker {
                    "(float, default 0.0) The value to pad for empty sequence.")
         .SetDefault(0.0);
     AddAttr<bool>("use_cvm", "bool, use cvm or not").SetDefault(true);
+    AddAttr<bool>("need_filter", "(bool, default false)").SetDefault(false);
+    AddAttr<float>("show_coeff", "(float, default 0.2)").SetDefault(0.2);
+    AddAttr<float>("clk_coeff", "(float, default 1)").SetDefault(1);
+    AddAttr<float>("threshold", "(float, default 0.96)").SetDefault(0.96);
 
     AddComment(R"DOC(
 Fuse multiple pairs of Sequence Pool and CVM Operator.