未验证 提交 7a4f09f1 编写于 作者: D duanyanhui 提交者: GitHub

fix save_combine_op (#49695)

上级 1bdb7960
......@@ -211,8 +211,8 @@ class SaveCombineOpKernel : public framework::OpKernel<T> {
if (inp_vars.size() > 0 && inp_vars[0]->IsType<phi::DenseTensor>()) {
std::vector<const phi::DenseTensor*> x(inp_vars.size());
for (auto inp_var : inp_vars) {
x.push_back(&(inp_var->Get<phi::DenseTensor>()));
for (size_t i = 0; i < inp_vars.size(); i++) {
x[i] = (&(inp_vars[i]->Get<phi::DenseTensor>()));
}
SaveCombineTensorKernel<T>(dev_ctx,
x,
......@@ -223,8 +223,8 @@ class SaveCombineOpKernel : public framework::OpKernel<T> {
output);
} else {
std::vector<const phi::ExtendedTensor*> x(inp_vars.size());
for (auto inp_var : inp_vars) {
x.push_back(&(inp_var->Get<framework::Vocab>()));
for (size_t i = 0; i < inp_vars.size(); i++) {
x[i] = (&(inp_vars[i]->Get<framework::Vocab>()));
}
SaveCombineVocabKernel<T>(dev_ctx,
x,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册