Merge pull request #16259 from phlrain/fix_lod_reset

phlrain · phlrain · commit e56058f1fb75 · 2019-03-22T07:40:14.000Z
Fix lod reset
diff --git a/paddle/fluid/operators/lod_reset_op.cc b/paddle/fluid/operators/lod_reset_op.cc
@@ -32,7 +32,10 @@ class LoDResetOp : public framework::OperatorWithKernel {
       PADDLE_ENFORCE_GT(level0.size(), 1,
                         "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"));
   }
 

Original file line number	Diff line number	Diff line change
`@@ -32,7 +32,10 @@ class LoDResetOp : public framework::OperatorWithKernel {`
`32`	`32`	`PADDLE_ENFORCE_GT(level0.size(), 1,`
`33`	`33`	`"If Input(Y) not provided, the target lod should be "`
`34`	`34`	"specified by attribute `target_lod`.");
	`35`	`+ } else {`
	`36`	`+ ctx->ShareLoD("Y", "Out");`
`35`	`37`	`}`
	`38`	`+`
`36`	`39`	`ctx->SetOutputDim("Out", ctx->GetInputDim("X"));`
`37`	`40`	`}`
`38`	`41`