Searched refs:ngram_size (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | skip_gram.cc | 62 return size <= params->ngram_size; in ShouldIncludeCurrentNgram()
64 return size == params->ngram_size; in ShouldIncludeCurrentNgram()
72 if (stack_idx < params->ngram_size && stack[stack_idx] + 1 < num_words) { in ShouldStepInRecursion()
109 if (words.size() < params->ngram_size) { in Eval()
116 std::vector<int> stack(params->ngram_size, 0); in Eval()
128 if (stack_idx < params->ngram_size) { in Eval()
|
| D | skip_gram_test.cc | 34 SkipGramOp(int ngram_size, int max_skip_size, bool include_all_ngrams) { in SkipGramOp() argument
39 CreateSkipGramOptions(builder_, ngram_size, max_skip_size, in SkipGramOp()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 255 int ngram_size; member
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 472 params->ngram_size = skip_gram_params->ngram_size(); in ParseOpData()
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_v2.fbs | 236 ngram_size: int;
|
| D | schema_v1.fbs | 228 ngram_size: int;
|
| D | schema_v3.fbs | 246 ngram_size: int;
|
| D | schema.fbs | 523 ngram_size: int;
|
| D | schema_generated.h | 4636 int32_t ngram_size;
4640 : ngram_size(0),
4653 int32_t ngram_size() const {
4677 void add_ngram_size(int32_t ngram_size) {
4678 fbb_.AddElement<int32_t>(SkipGramOptions::VT_NGRAM_SIZE, ngram_size, 0);
4700 int32_t ngram_size = 0,
4705 builder_.add_ngram_size(ngram_size);
9668 { auto _e = ngram_size(); _o->ngram_size = _e; };
9681 auto _ngram_size = _o->ngram_size;
|
| /external/tensorflow/tensorflow/lite/models/smartreply/g3doc/ |
| D | README.md | 117 skip grams. The configurable parameters are `ngram_size` and
|