add ShareLoD for dropout_grad (#14616)

test=develop
7 years ago · 6648f5ed6f
parent 3c77ce3751
commit 6648f5ed6f
1 changed files with 1 additions and 0 deletions
--- a/paddle/fluid/operators/dropout_op.cc
+++ b/paddle/fluid/operators/dropout_op.cc
@ -120,6 +120,7 @@ class DropoutOpGrad : public framework::OperatorWithKernel {
                      "Dimensions of Input(X) and Mask must be the same.");

    ctx->SetOutputDim(framework::GradVarName("X"), x_dims);
+    ctx->ShareLoD("X", /*->*/ framework::GradVarName("X"));
  }
 };