Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 1 年 前同步成功

通知 2298
Star 20931
Fork 5422
P
Paddle

体验新版 GitCode，发现更多精彩内容 >>

未验证 提交 3fb4a08c 编写于 作者: X xiaoxiaohehe001 提交者： GitHub
浏览文件
操作

fix_split (#49743)

上级 adcb0039
隐藏空白更改
内联 并排
Showing with 3 addition and 3 deletion
paddle/phi/infermeta/unary.cc
浏览文件 @ 3fb4a08c
...@@ -3437,11 +3437,11 @@ void FillSplitOutDims(const MetaTensor& x, ...@@ -3437,11 +3437,11 @@ void FillSplitOutDims(const MetaTensor& x,
(*out)[i]->set_dtype(x.dtype()); (*out)[i]->set_dtype(x.dtype());
(*out)[i]->set_dims(out_dims[i]); (*out)[i]->set_dims(out_dims[i]);
(*out)[i]->set_layout(x.layout()); (*out)[i]->set_layout(x.layout());
(*out)[i]->share_lod(x);
} else { } else {
(*out)[i]->set_dtype(x.dtype()); (*out)[i]->set_dtype(x.dtype());
(*out)[i]->set_dims(out_dims[i]); (*out)[i]->set_dims(out_dims[i]);
(*out)[i]->set_layout(x.layout()); (*out)[i]->set_layout(x.layout());
(*out)[i]->share_lod(x);
} }
} }
} }
...@@ -3472,11 +3472,11 @@ void SplitInferMeta(const MetaTensor& x, ...@@ -3472,11 +3472,11 @@ void SplitInferMeta(const MetaTensor& x,
out[i]->set_dtype(x.dtype()); out[i]->set_dtype(x.dtype());
out[i]->set_dims(out_dims[i]); out[i]->set_dims(out_dims[i]);
out[i]->set_layout(x.layout()); out[i]->set_layout(x.layout());
out[i]->share_lod(x);
} else { } else {
out[i]->set_dtype(x.dtype()); out[i]->set_dtype(x.dtype());
out[i]->set_dims(out_dims[i]); out[i]->set_dims(out_dims[i]);
out[i]->set_layout(x.layout()); out[i]->set_layout(x.layout());
out[i]->share_lod(x);
} }
} }
} else { } else {
...@@ -3563,11 +3563,11 @@ void SplitWithNumInferMeta(const MetaTensor& x, ...@@ -3563,11 +3563,11 @@ void SplitWithNumInferMeta(const MetaTensor& x,
out[i]->set_dtype(x.dtype()); out[i]->set_dtype(x.dtype());
out[i]->set_dims(out_dims[i]); out[i]->set_dims(out_dims[i]);
out[i]->set_layout(x.layout()); out[i]->set_layout(x.layout());
out[i]->share_lod(x);
} else { } else {
out[i]->set_dtype(x.dtype()); out[i]->set_dtype(x.dtype());
out[i]->set_dims(out_dims[i]); out[i]->set_dims(out_dims[i]);
out[i]->set_layout(x.layout()); out[i]->set_layout(x.layout());
out[i]->share_lod(x);
} }
} }
} else { } else {
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或