diff --git a/paddle/fluid/operators/dropout_op.cc b/paddle/fluid/operators/dropout_op.cc index dd3474dd2529b5e2cb2cd32aec41fb6357b5d537..2ccc86c1dc04a3afeb02b24677e6ebce40cca4fa 100644 --- 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")); } };