diff --git a/paddle/fluid/framework/ir/seqpool_cvm_concat_fuse_pass.cc b/paddle/fluid/framework/ir/seqpool_cvm_concat_fuse_pass.cc index 50364d2d0109d..55ef6a492954e 100644 --- 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()