| 451 | } |
| 452 | |
| 453 | void GetInputPerBatch(const std::vector<std::vector<int64_t>> &in, |
| 454 | std::vector<std::vector<int64_t>> *out, |
| 455 | std::vector<size_t> *lod, |
| 456 | size_t batch_iter, |
| 457 | size_t batch_end) { |
| 458 | lod->clear(); |
| 459 | lod->push_back(0); |
| 460 | for (auto it = in.begin() + batch_iter; it < in.begin() + batch_end; it++) { |
| 461 | out->push_back(*it); |
| 462 | lod->push_back(lod->back() + (*it).size()); // calculate lod |
| 463 | } |
| 464 | } |
| 465 | |
| 466 | void ConvertPaddleTensorToZeroCopyTensor( |
| 467 | PaddlePredictor *predictor, const std::vector<PaddleTensor> &inputs) { |