Home
last modified time | relevance | path

Searched refs:ngram_size (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dskip_gram.cc63 return size <= params->ngram_size; in ShouldIncludeCurrentNgram()
65 return size == params->ngram_size; in ShouldIncludeCurrentNgram()
73 if (stack_idx < params->ngram_size && stack[stack_idx] + 1 < num_words) { in ShouldStepInRecursion()
110 if (words.size() < params->ngram_size) { in Eval()
117 std::vector<int> stack(params->ngram_size, 0); in Eval()
129 if (stack_idx < params->ngram_size) { in Eval()
Dskip_gram_test.cc34 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/core/kernels/
Dstring_ngrams_op.cc144 int ngram_size = 0; in CreateNgrams() local
146 ngram_size += left_padding * left_pad_.length(); in CreateNgrams()
149 ngram_size += data[data_start_index + n].length(); in CreateNgrams()
152 ngram_size += right_padding * right_pad_.length(); in CreateNgrams()
155 ngram_size += num_separators * separator_.length(); in CreateNgrams()
159 ngram->reserve(ngram_size); in CreateNgrams()
176 DCHECK_EQ(ngram_size, ngram->size()); in CreateNgrams()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h286 int ngram_size; member
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc525 params->ngram_size = skip_gram_params->ngram_size(); in ParseOpData()
/external/tensorflow/tensorflow/lite/schema/
Dschema_v1.fbs228 ngram_size: int;
Dschema_v2.fbs236 ngram_size: int;
Dschema_v3.fbs246 ngram_size: int;
Dschema.fbs643 ngram_size: int;
Dschema_generated.h5227 int32_t ngram_size;
5231 : ngram_size(0),
5244 int32_t ngram_size() const {
5268 void add_ngram_size(int32_t ngram_size) {
5269 fbb_.AddElement<int32_t>(SkipGramOptions::VT_NGRAM_SIZE, ngram_size, 0);
5291 int32_t ngram_size = 0,
5296 builder_.add_ngram_size(ngram_size);
11140 { auto _e = ngram_size(); _o->ngram_size = _e; };
11153 auto _ngram_size = _o->ngram_size;