Searched refs:num_pieces (Results 1 – 4 of 4) sorted by relevance
| /external/libtextclassifier/utils/sentencepiece/ |
| D | encoder.cc | 51 /*num_pieces=*/segmentation[i].num_pieces}; in Encode()
56 /*num_pieces=*/segmentation[i].num_pieces + 1}; in Encode()
74 /*num_pieces=*/segmentation[i].num_pieces + 1}; in Encode()
80 if (segmentation[len].num_pieces <= 0) { in Encode()
84 const int num_pieces = segmentation[len].num_pieces; in Encode() local
85 encoded_text->resize(num_pieces + 2); in Encode()
86 (*encoded_text)[num_pieces + 1] = end_code_; in Encode()
88 for (int i = num_pieces; i > 0; i--) { in Encode()
|
| D | encoder.h | 42 Encoder(const SentencePieceMatcher* matcher, const int num_pieces,
46 : num_pieces_(num_pieces), in num_pieces_() argument
74 int num_pieces; member
|
| D | sorted_strings_table.h | 40 SortedStringsTable(const int num_pieces, const uint32* offsets,
43 : num_pieces_(num_pieces), in num_pieces_() argument
|
| /external/libtextclassifier/utils/tflite/ |
| D | text_encoder.cc | 102 const int num_pieces = config->pieces_scores()->Length(); in Initialize() local
116 num_pieces, config->pieces_offsets()->data(), in Initialize()
126 encoder_op->matcher.get(), num_pieces, config->pieces_scores()->data(), in Initialize()
|