fix bugs, test=develop

PaddlePaddle · Jun 22, 2021 · 019e443 · 019e443
1 parent 41bfad5
commit 019e443
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/paddle/fluid/framework/ir/seqpool_cvm_concat_fuse_pass.cc b/paddle/fluid/framework/ir/seqpool_cvm_concat_fuse_pass.cc
@@ -61,8 +61,10 @@ SeqPoolCVMConcatFusePass::SeqPoolCVMConcatFusePass() {
       .IsTensor()
       .End()
       .AddOutput("MaxIndex")
+      .IsOptional()
       .End()
       .AddAttr("pooltype")
+      .IsStringIn({"AVERAGE", "SUM", "SQRT", "LAST", "FIRST", "MAX"})
       .End()
       .AddAttr("pad_value")
       .End();
@@ -85,6 +87,7 @@ SeqPoolCVMConcatFusePass::SeqPoolCVMConcatFusePass() {
       .End()
       .AddInput("AxisTensor")
       .IsTensor()
+      .IsOptional()
       .End()
       .AddOutput("Out")
       .IsTensor()