diff --git a/paddle/fluid/operators/sequence_ops/sequence_unpad_op.cc b/paddle/fluid/operators/sequence_ops/sequence_unpad_op.cc index 23581c360ff541e6d9d1000e9b2bae3712e15dd9..8256460858b88bf30e4d70e796eebc84bd68c0da 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(ctx->GetOutputVarPtrs("Out")[0]); + out_desc->SetLoDLevel(1); + } } protected: