diff --git a/paddle/phi/infermeta/unary.cc b/paddle/phi/infermeta/unary.cc index da24fed946bb164fdd47d299c104d8ba82fa0b5a..04373fa29edf9e58661d8c52cf85d08be0ceaf13 100644 --- a/paddle/phi/infermeta/unary.cc +++ b/paddle/phi/infermeta/unary.cc @@ -3437,11 +3437,11 @@ void FillSplitOutDims(const MetaTensor& x, (*out)[i]->set_dtype(x.dtype()); (*out)[i]->set_dims(out_dims[i]); (*out)[i]->set_layout(x.layout()); + (*out)[i]->share_lod(x); } else { (*out)[i]->set_dtype(x.dtype()); (*out)[i]->set_dims(out_dims[i]); (*out)[i]->set_layout(x.layout()); - (*out)[i]->share_lod(x); } } } @@ -3472,11 +3472,11 @@ void SplitInferMeta(const MetaTensor& x, out[i]->set_dtype(x.dtype()); out[i]->set_dims(out_dims[i]); out[i]->set_layout(x.layout()); + out[i]->share_lod(x); } else { out[i]->set_dtype(x.dtype()); out[i]->set_dims(out_dims[i]); out[i]->set_layout(x.layout()); - out[i]->share_lod(x); } } } else { @@ -3563,11 +3563,11 @@ void SplitWithNumInferMeta(const MetaTensor& x, out[i]->set_dtype(x.dtype()); out[i]->set_dims(out_dims[i]); out[i]->set_layout(x.layout()); + out[i]->share_lod(x); } else { out[i]->set_dtype(x.dtype()); out[i]->set_dims(out_dims[i]); out[i]->set_layout(x.layout()); - out[i]->share_lod(x); } } } else {