未验证 提交 3fb4a08c 编写于 作者: X xiaoxiaohehe001 提交者: GitHub

fix_split (#49743)

上级 adcb0039
......@@ -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 {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册