Searched refs:TfLiteSkipGramParams (Results 1 – 4 of 4) sorted by relevance
57 bool ShouldIncludeCurrentNgram(const TfLiteSkipGramParams* params, int size) { in ShouldIncludeCurrentNgram()68 bool ShouldStepInRecursion(const TfLiteSkipGramParams* params, in ShouldStepInRecursion()89 auto* params = reinterpret_cast<TfLiteSkipGramParams*>(node->builtin_data); in Eval()
57 TfLiteSkipGramParams skip_gram_params; in TEST()
258 } TfLiteSkipGramParams; typedef
469 TfLiteSkipGramParams* params = in ParseOpData()470 allocator->AllocatePOD<TfLiteSkipGramParams>(); in ParseOpData()