From a21fdde218b0ed95b43afc480eea6e819d4b5663 Mon Sep 17 00:00:00 2001 From: phlrain Date: Mon, 18 Mar 2019 09:24:56 +0000 Subject: [PATCH] fix lod reset op;test=develop --- paddle/fluid/operators/lod_reset_op.cc | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/paddle/fluid/operators/lod_reset_op.cc b/paddle/fluid/operators/lod_reset_op.cc index 7c8fe5fbd76..f71463a103a 100644 --- a/paddle/fluid/operators/lod_reset_op.cc +++ b/paddle/fluid/operators/lod_reset_op.cc @@ -33,6 +33,11 @@ class LoDResetOp : public framework::OperatorWithKernel { "If Input(Y) not provided, the target lod should be " "specified by attribute `target_lod`."); } + else + { + ctx->ShareLoD( "Y", "Out"); + } + ctx->SetOutputDim("Out", ctx->GetInputDim("X")); } -- GitLab