add lod for slice op, test=develop

0d4b60ab · jerrywgz · 66bb5dd7 · 0d4b60ab
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

paddle/fluid/operators/slice_op.cc paddle/fluid/operators/slice_op.cc +3 -0

未找到文件。
--- 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: