| 55 | } |
| 56 | |
| 57 | bool ShouldIncludeCurrentNgram(const TfLiteSkipGramParams* params, int size) { |
| 58 | if (size <= 0) { |
| 59 | return false; |
| 60 | } |
| 61 | if (params->include_all_ngrams) { |
| 62 | return size <= params->ngram_size; |
| 63 | } else { |
| 64 | return size == params->ngram_size; |
| 65 | } |
| 66 | } |
| 67 | |
| 68 | bool ShouldStepInRecursion(const TfLiteSkipGramParams* params, |
| 69 | const std::vector<int>& stack, int stack_idx, |