未验证 提交 6648f5ed 编写于 作者: C chengduo 提交者: GitHub

add ShareLoD for dropout_grad (#14616)

test=develop
上级 3c77ce37
...@@ -120,6 +120,7 @@ class DropoutOpGrad : public framework::OperatorWithKernel { ...@@ -120,6 +120,7 @@ class DropoutOpGrad : public framework::OperatorWithKernel {
"Dimensions of Input(X) and Mask must be the same."); "Dimensions of Input(X) and Mask must be the same.");
ctx->SetOutputDim(framework::GradVarName("X"), x_dims); ctx->SetOutputDim(framework::GradVarName("X"), x_dims);
ctx->ShareLoD("X", /*->*/ framework::GradVarName("X"));
} }
}; };
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册