add lod for slice op, test=develop

6 years ago · 0d4b60ab8b
parent 66bb5dd760
commit 0d4b60ab8b
1 changed files with 3 additions and 0 deletions
--- a/paddle/fluid/operators/slice_op.cc
+++ b/paddle/fluid/operators/slice_op.cc
@ -54,6 +54,9 @@ class SliceOp : public framework::OperatorWithKernel {
      out_dims[axes[i]] = end - start;
    }
    ctx->SetOutputDim("Out", out_dims);
+    if (axes[0] != 0) {
+      ctx->ShareLoD("Input", /*->*/ "Out");
+    }
  }

 protected: