test=develop

local_add_cudnn_lstm
JiabinYang 6 years ago
parent f7b55de9e5
commit 57a18e32a1

@ -43,9 +43,7 @@ std::vector<int64_t> cal_rows(const framework::LoDTensor* path) {
}
}
}
for (std::set<int64_t>::iterator it = tmp.begin(); it != tmp.end(); ++it) {
rows.push_back(*it);
}
rows.assign(tmp.begin(), tmp.end());
return rows;
}

Loading…
Cancel
Save