Home
last modified time | relevance | path

Searched refs:stack_idx (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dskip_gram.cc73 const std::vector<int>& stack, int stack_idx, in ShouldStepInRecursion() argument
76 if (stack_idx < params->ngram_size && stack[stack_idx] + 1 < num_words) { in ShouldStepInRecursion()
78 if (stack_idx == 0) { in ShouldStepInRecursion()
85 if (stack[stack_idx] - stack[stack_idx - 1] <= params->max_skip_size) { in ShouldStepInRecursion()
124 int stack_idx = 1; in Eval() local
127 while (stack_idx >= 0) { in Eval()
128 if (ShouldStepInRecursion(params, stack, stack_idx, num_words)) { in Eval()
132 stack[stack_idx]++; in Eval()
133 stack_idx++; in Eval()
134 if (stack_idx < params->ngram_size) { in Eval()
[all …]
/external/libtextclassifier/native/utils/tflite/
Dstring_projection.cc385 bool ShouldStepInRecursion(const std::vector<int>& stack, int stack_idx, in ShouldStepInRecursion() argument
388 if (stack_idx < params.ngram_size && stack[stack_idx] + 1 < num_words) { in ShouldStepInRecursion()
390 if (stack_idx == 0) { in ShouldStepInRecursion()
397 if (stack[stack_idx] - stack[stack_idx - 1] <= params.max_skip_size) { in ShouldStepInRecursion()
404 std::string JoinTokensBySpace(const std::vector<int>& stack, int stack_idx, in JoinTokensBySpace() argument
407 for (int i = 0; i < stack_idx; i++) { in JoinTokensBySpace()
410 len += stack_idx - 1; in JoinTokensBySpace()
415 for (int i = 1; i < stack_idx; i++) { in JoinTokensBySpace()
438 int stack_idx = 1; in ExtractSkipGramsImpl() local
441 while (stack_idx >= 0) { in ExtractSkipGramsImpl()
[all …]
/external/perfetto/src/trace_processor/importers/common/
Dslice_tracker.cc110 base::Optional<uint32_t> stack_idx = in AddArgs() local
112 if (!stack_idx.has_value()) in AddArgs()
114 uint32_t slice_idx = stack[*stack_idx].row; in AddArgs()
117 ArgsTracker* tracker = &stack[*stack_idx].args_tracker; in AddArgs()
208 base::Optional<uint32_t> stack_idx = finder(stack); in CompleteSlice() local
212 if (!stack_idx) in CompleteSlice()
215 const auto& slice_info = stack[stack_idx.value()]; in CompleteSlice()
220 ArgsTracker* tracker = &stack[stack_idx.value()].args_tracker; in CompleteSlice()
238 if (*stack_idx == stack.size() - 1) in CompleteSlice()
/external/libaom/libaom/av1/common/
Dmvref_common.c454 int stack_idx; in process_single_ref_mv_candidate() local
455 for (stack_idx = 0; stack_idx < *refmv_count; ++stack_idx) { in process_single_ref_mv_candidate()
456 const int_mv stack_mv = ref_mv_stack[stack_idx].this_mv; in process_single_ref_mv_candidate()
460 if (stack_idx == *refmv_count) { in process_single_ref_mv_candidate()
461 ref_mv_stack[stack_idx].this_mv = this_mv; in process_single_ref_mv_candidate()
465 ref_mv_weight[stack_idx] = 2; in process_single_ref_mv_candidate()