Searched refs:num_pieces (Results 1 – 4 of 4) sorted by relevance
/external/libtextclassifier/native/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 StringSet* pieces, const int num_pieces, 46 : num_pieces_(num_pieces), in num_pieces_() argument 74 int num_pieces; member
|
/external/libtextclassifier/native/utils/container/ |
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/native/utils/tflite/ |
D | text_encoder.cc | 103 const int num_pieces = config->pieces_scores()->size(); in Initialize() local 117 num_pieces, config->pieces_offsets()->data(), in Initialize() 127 encoder_op->matcher.get(), num_pieces, config->pieces_scores()->data(), in Initialize()
|