Home
last modified time | relevance | path

Searched refs:model_spec (Results 1 – 5 of 5) sorted by relevance

/external/libtextclassifier/actions/
Dlua-actions.cc53 const TensorflowLiteModelSpec* model_spec, in CreateLuaActionsSuggestions() argument
59 snippet, conversation, model_executor, model_spec, interpreter, in CreateLuaActionsSuggestions()
72 const TensorflowLiteModelSpec* model_spec, in LuaActionsSuggestions() argument
80 model_spec == nullptr in LuaActionsSuggestions()
83 model_spec->output_actions_scores())), in LuaActionsSuggestions()
85 model_spec == nullptr in LuaActionsSuggestions()
88 model_spec->output_replies_scores())), in LuaActionsSuggestions()
89 sensitivity_score_(model_spec == nullptr in LuaActionsSuggestions()
93 model_spec->output_sensitive_topic_score())), in LuaActionsSuggestions()
95 model_spec == nullptr in LuaActionsSuggestions()
[all …]
Dlua-actions.h35 const TensorflowLiteModelSpec* model_spec,
55 const TensorflowLiteModelSpec* model_spec,
/external/libtextclassifier/annotator/
Dmodel-executor.h35 const tflite::Model* model_spec) { in FromModelSpec() argument
36 auto model = TfLiteModelFromModelSpec(model_spec); in FromModelSpec()
/external/libtextclassifier/utils/
Dtflite-model-executor.h44 const tflite::Model* model_spec) { in FromModelSpec() argument
45 auto model = TfLiteModelFromModelSpec(model_spec); in FromModelSpec()
Dtflite-model-executor.cc179 const tflite::Model* model_spec) { in TfLiteModelFromModelSpec() argument
181 tflite::FlatBufferModel::BuildFromModel(model_spec)); in TfLiteModelFromModelSpec()