未验证 提交 9874530e 编写于 作者: H Hongyu Liu 提交者: GitHub

Update lod_reset_op.cc

上级 58c69a9e
...@@ -33,7 +33,7 @@ class LoDResetOp : public framework::OperatorWithKernel { ...@@ -33,7 +33,7 @@ class LoDResetOp : public framework::OperatorWithKernel {
"If Input(Y) not provided, the target lod should be " "If Input(Y) not provided, the target lod should be "
"specified by attribute `target_lod`."); "specified by attribute `target_lod`.");
} else { } else {
ctx->ShareLoD( "Y", "Out"); ctx->ShareLoD("Y", "Out");
} }
ctx->SetOutputDim("Out", ctx->GetInputDim("X")); ctx->SetOutputDim("Out", ctx->GetInputDim("X"));
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册