Searched refs:TfLiteSkipGramParams (Results 1 – 4 of 4) sorted by relevance
61 bool ShouldIncludeCurrentNgram(const TfLiteSkipGramParams* params, int size) { in ShouldIncludeCurrentNgram()72 bool ShouldStepInRecursion(const TfLiteSkipGramParams* params, in ShouldStepInRecursion()93 auto* params = reinterpret_cast<TfLiteSkipGramParams*>(node->builtin_data); in Eval()
57 TfLiteSkipGramParams skip_gram_params; in TEST()
341 } TfLiteSkipGramParams; typedef
628 auto params = safe_allocator.Allocate<TfLiteSkipGramParams>(); in ParseOpDataTfLite()