From 3ec48480af191970ad12377c9161994920fbf722 Mon Sep 17 00:00:00 2001
From: dangqingqing <dangqingqing@baidu.com>
Date: Wed, 20 Sep 2017 19:05:36 +0800
Subject: [PATCH] Fix bug

---
 paddle/operators/accuracy_op.cc | 6 +++---
 paddle/operators/dropout_op.cc  | 1 +
 paddle/operators/fc_op.cc       | 1 -
 paddle/operators/pad_op.cc      | 5 +++++
 paddle/operators/reshape_op.cc  | 5 +++++
 5 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/paddle/operators/accuracy_op.cc b/paddle/operators/accuracy_op.cc
index 32479ae5a35..391258b40ba 100644
--- a/paddle/operators/accuracy_op.cc
+++ b/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");
   }
 };
 
diff --git a/paddle/operators/dropout_op.cc b/paddle/operators/dropout_op.cc
index b111b9fccb2..bfa1992d79f 100644
--- a/paddle/operators/dropout_op.cc
+++ b/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");
   }
 };
 
diff --git a/paddle/operators/fc_op.cc b/paddle/operators/fc_op.cc
index 56fe654d1e8..5ac0e8cc45f 100644
--- a/paddle/operators/fc_op.cc
+++ b/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");
   }
 };
diff --git a/paddle/operators/pad_op.cc b/paddle/operators/pad_op.cc
index a0b1c6b631d..98de18fb9f6 100644
--- a/paddle/operators/pad_op.cc
+++ b/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");
+    }
   }
 };
 
diff --git a/paddle/operators/reshape_op.cc b/paddle/operators/reshape_op.cc
index 0d05e344148..c0907586199 100644
--- a/paddle/operators/reshape_op.cc
+++ b/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");
+    }
   }
 };
 
-- 
GitLab