Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 0d4b60ab 编写于 作者: J jerrywgz
浏览文件
操作

add lod for slice op, test=develop

上级 66bb5dd7
显示空白变更内容
内联 并排
Showing with 3 addition and 0 deletion
paddle/fluid/operators/slice_op.cc
浏览文件 @ 0d4b60ab
...@@ -54,6 +54,9 @@ class SliceOp : public framework::OperatorWithKernel { ...@@ -54,6 +54,9 @@ class SliceOp : public framework::OperatorWithKernel {
out_dims[axes[i]] = end - start; out_dims[axes[i]] = end - start;
} }
ctx->SetOutputDim("Out", out_dims); ctx->SetOutputDim("Out", out_dims);
if (axes[0] != 0) {
ctx->ShareLoD("Input", /*->*/ "Out");
}
} }
protected: protected:
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或