Fix bug

paddle/operators/accuracy_op.cc

@@ -55,15 +55,15 @@ class AccuracyOpMaker : public framework::OpProtoAndCheckerMaker {
     // TODO(typhoonzero): AddInput("Weight", ...
     AddOutput("Accuracy", "The accuracy of current batch");
 
-    AddComment(
-        R"DOC(Accuracy. It will print accuracy rate for classification.
+    AddComment(R"DOC(
+Accuracy. It will print accuracy rate for classification.
 The accuracy is:
 ..  math::
 accuracy = \\frac{NumOfCorrectPredicts}{NumOfAllSamples})
 
 Both the input `Inference` and `Label` can carry the LoD (Level of Details)
 information, or not. But the output only shares the LoD with input `Inference`.
-DOC");
+)DOC");
   }
 };
 

paddle/operators/dropout_op.cc

@@ -38,6 +38,7 @@ class DropoutOp : public framework::OperatorWithKernel {
     if (ctx.Attr<int>("is_training") == 1) {
       ctx.Output<LoDTensor>("Mask")->Resize(dims);
     }
+    ctx.ShareLoD("X", "Out");
   }
 };
 

paddle/operators/fc_op.cc

@@ -189,7 +189,6 @@ Activation type can be set to `identity` (default), `sigmoid` or `softmax`.
 
 All the inputs can carry the LoD (Level of Details) information,
 or not. But the output only shares the LoD with first input (`X[0]`).
-)DOC");
 )DOC");
   }
 };

paddle/operators/pad_op.cc

@@ -41,6 +41,11 @@ class PadOp : public framework::OperatorWithKernel {
     }
     ctx.Output<framework::LoDTensor>("Out")->Resize(
         framework::make_ddim(out_dims));
+    if (out_dims[0] == x_dim[0]) {
+      // Only pass LoD when the first dimension is equal between
+      // output and input.
+      ctx.ShareLoD("X", "Out");
+    }
   }
 };
 

paddle/operators/reshape_op.cc

@@ -51,6 +51,11 @@ class ReshapeOp : public framework::OperatorWithKernel {
                    [](int a) { return static_cast<int64_t>(a); });
     auto out_dims = framework::make_ddim(shape_int64);
     ctx.Output<framework::LoDTensor>("Out")->Resize(out_dims);
+    if (shape[0] == in->dims()[0]) {
+      // Only pass LoD when the first dimension is equal between
+      // output and input.
+      ctx.ShareLoD("X", "Out");
+    }
   }
 };