提交 a21fdde2 编写于 作者: P phlrain

fix lod reset op;test=develop

上级 dcba2e72
...@@ -33,6 +33,11 @@ class LoDResetOp : public framework::OperatorWithKernel { ...@@ -33,6 +33,11 @@ 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
{
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.
先完成此消息的编辑!
想要评论请 注册