| 432 | } |
| 433 | |
| 434 | util::Status SentencePieceProcessor::NBestEncode( |
| 435 | absl::string_view input, int nbest_size, |
| 436 | std::vector<std::vector<std::string>> *pieces) const { |
| 437 | CHECK_OR_RETURN_STATUS_STL(pieces); |
| 438 | |
| 439 | NBestSentencePieceText spt; |
| 440 | RETURN_IF_ERROR(NBestEncode(input, nbest_size, &spt)); |
| 441 | pieces->reserve(spt.nbests().size()); |
| 442 | for (const auto &nbest : spt.nbests()) { |
| 443 | std::vector<std::string> &result = pieces->emplace_back(); |
| 444 | result.reserve(nbest.pieces().size()); |
| 445 | for (const auto &sp : nbest.pieces()) { |
| 446 | result.emplace_back(sp.piece()); |
| 447 | } |
| 448 | } |
| 449 | |
| 450 | return util::OkStatus(); |
| 451 | } |
| 452 | |
| 453 | util::Status SentencePieceProcessor::NBestEncode( |
| 454 | absl::string_view input, int nbest_size, |
no test coverage detected