diff --git a/paddle/fluid/operators/sequence_ops/sequence_unpad_op.cc b/paddle/fluid/operators/sequence_ops/sequence_unpad_op.cc index 23581c360f..8256460858 100644 --- a/paddle/fluid/operators/sequence_ops/sequence_unpad_op.cc +++ b/paddle/fluid/operators/sequence_ops/sequence_unpad_op.cc @@ -57,6 +57,11 @@ class SequenceUnpadOp : public framework::OperatorWithKernel { } } ctx->SetOutputDim("Out", framework::make_ddim(out_dims_vec)); + if (!ctx->IsRuntime()) { + framework::VarDesc* out_desc = + boost::get<framework::VarDesc*>(ctx->GetOutputVarPtrs("Out")[0]); + out_desc->SetLoDLevel(1); + } } protected: