We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent dddc5d9 commit e32ff65Copy full SHA for e32ff65
paddle/fluid/framework/ir/constant_folding_pass.cc
@@ -142,6 +142,9 @@ void ConstantFoldingPass::ApplyImpl(ir::Graph *graph) const {
142
}
143
out_desc->SetShape(out_shape);
144
out_desc->SetPersistable(true);
145
+ auto *var_desc_out = op_node->Op()->Block()->Var(out_name);
146
+ var_desc_out->SetShape(out_shape);
147
+ var_desc_out->SetPersistable(true);
148
auto *global_out_tensor = scope->Var(out_name)->GetMutable<LoDTensor>();
149
*global_out_tensor = *local_out_tensor;
150
0 commit comments