diff --git a/paddle/fluid/operators/gather_nd_op.cc b/paddle/fluid/operators/gather_nd_op.cc
index f871615d0bda2572e6cbf8cf8cfe0a5315247eae..3ba5beec3bb87ed5b08756c0779560525c666da7 100644
--- a/paddle/fluid/operators/gather_nd_op.cc
+++ b/paddle/fluid/operators/gather_nd_op.cc
@@ -55,6 +55,7 @@ class GatherNdOp : public framework::OperatorWithKernel {
     }
 
     ctx->SetOutputDim("Out", framework::make_ddim(result_dims));
+    ctx->ShareLoD("X", /*->*/ "Out");
   }
 
  protected:
diff --git a/paddle/fluid/operators/scatter_nd_add_op.cc b/paddle/fluid/operators/scatter_nd_add_op.cc
index e86724dd569199ecb87f52ea29268bfc68ed5957..11cf01c224c818625e239b59133ff4f9d722c264 100644
--- a/paddle/fluid/operators/scatter_nd_add_op.cc
+++ b/paddle/fluid/operators/scatter_nd_add_op.cc
@@ -64,6 +64,7 @@ class ScatterNdAddOp : public framework::OperatorWithKernel {
                         "Updates has wrong shape");
     }
     ctx->SetOutputDim("Out", ref_dims);
+    ctx->ShareLoD("X", /*->*/ "Out");
   }
 
  protected: