Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dskip_gram.cc57 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()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data_test.cc57 TfLiteSkipGramParams skip_gram_params; in TEST()
Dbuiltin_op_data.h258 } TfLiteSkipGramParams; typedef
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc469 TfLiteSkipGramParams* params = in ParseOpData()
470 allocator->AllocatePOD<TfLiteSkipGramParams>(); in ParseOpData()