Searched refs:ngram_size (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | skip_gram.cc | 66 return size <= params->ngram_size; in ShouldIncludeCurrentNgram()
68 return size == params->ngram_size; in ShouldIncludeCurrentNgram()
76 if (stack_idx < params->ngram_size && stack[stack_idx] + 1 < num_words) { in ShouldStepInRecursion()
115 if (words.size() < params->ngram_size) { in Eval()
122 std::vector<int> stack(params->ngram_size, 0); in Eval()
134 if (stack_idx < params->ngram_size) { in Eval()
|
| D | skip_gram_test.cc | 37 SkipGramOp(int ngram_size, int max_skip_size, bool include_all_ngrams) { in SkipGramOp() argument
42 CreateSkipGramOptions(builder_, ngram_size, max_skip_size, in SkipGramOp()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | string_ngrams_op.cc | 157 int ngram_size = 0; in CreateNgrams() local
159 ngram_size += left_padding * left_pad_.length(); in CreateNgrams()
162 ngram_size += data[data_start_index + n].length(); in CreateNgrams()
165 ngram_size += right_padding * right_pad_.length(); in CreateNgrams()
168 ngram_size += num_separators * separator_.length(); in CreateNgrams()
172 ngram->reserve(ngram_size); in CreateNgrams()
189 DCHECK_EQ(ngram_size, ngram->size()); in CreateNgrams()
|
| /external/libtextclassifier/native/utils/tflite/ |
| D | string_projection.cc | 379 return size <= params.ngram_size; in ShouldIncludeCurrentNgram()
381 return size == params.ngram_size; in ShouldIncludeCurrentNgram()
388 if (stack_idx < params.ngram_size && stack[stack_idx] + 1 < num_words) { in ShouldStepInRecursion()
436 std::vector<int> stack(params.ngram_size + 1, 0); in ExtractSkipGramsImpl()
|
| D | string_projection_base.h | 42 int ngram_size; member
|
| D | string_projection_base.cc | 61 .ngram_size = custom_options["ngram_size"].AsInt32(), in StringProjectionOpBase()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 338 int ngram_size; member
|
| /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_v3a.fbs | 696 ngram_size: int;
|
| D | schema.fbs | 724 ngram_size: int;
|
| D | schema_generated.h | 6009 int32_t ngram_size;
6013 : ngram_size(0),
6026 int32_t ngram_size() const {
6050 void add_ngram_size(int32_t ngram_size) {
6051 fbb_.AddElement<int32_t>(SkipGramOptions::VT_NGRAM_SIZE, ngram_size, 0);
6073 int32_t ngram_size = 0,
6078 builder_.add_ngram_size(ngram_size);
12636 { auto _e = ngram_size(); _o->ngram_size = _e; }
12649 auto _ngram_size = _o->ngram_size;
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 632 params->ngram_size = skip_gram_params->ngram_size(); in ParseOpDataTfLite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | test_schema.fbs | 689 ngram_size: int;
|