Home
last modified time | relevance | path

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

/external/libtextclassifier/native/actions/
Dlua-actions.cc58 const TensorflowLiteModelSpec* model_spec, in CreateLuaActionsSuggestions() argument
64 snippet, conversation, model_executor, model_spec, interpreter, in CreateLuaActionsSuggestions()
77 const TensorflowLiteModelSpec* model_spec, in LuaActionsSuggestions() argument
84 model_spec == nullptr in LuaActionsSuggestions()
87 model_spec->output_actions_scores())), in LuaActionsSuggestions()
89 model_spec == nullptr in LuaActionsSuggestions()
92 model_spec->output_replies_scores())), in LuaActionsSuggestions()
93 sensitivity_score_(model_spec == nullptr in LuaActionsSuggestions()
97 model_spec->output_sensitive_topic_score())), in LuaActionsSuggestions()
99 model_spec == nullptr in LuaActionsSuggestions()
[all …]
Dtflite-sensitive-model.cc78 if (model_config_->model_spec()->input_context() >= 0) { in EvalConversation()
79 if (model_config_->model_spec()->input_length_to_pad() > 0) { in EvalConversation()
80 context.resize(model_config_->model_spec()->input_length_to_pad()); in EvalConversation()
83 model_config_->model_spec()->input_context(), context, in EvalConversation()
86 if (model_config_->model_spec()->input_context_length() >= 0) { in EvalConversation()
88 model_config_->model_spec()->input_context_length(), context.size(), in EvalConversation()
93 if (model_config_->model_spec()->input_num_suggestions() > 0) { in EvalConversation()
95 model_config_->model_spec()->input_num_suggestions(), 3, in EvalConversation()
107 model_config_->model_spec()->output_replies(), interpreter.get()); in EvalConversation()
109 model_config_->model_spec()->output_replies_scores(), interpreter.get()); in EvalConversation()
[all …]
Dlua-actions.h34 const TensorflowLiteModelSpec* model_spec,
45 const TensorflowLiteModelSpec* model_spec,
Dactions_model.fbs253 model_spec:TensorflowLiteModelSpec;
/external/tensorflow/tensorflow/lite/g3doc/tutorials/
Dmodel_maker_text_classification.ipynb131 "from tflite_model_maker import model_spec\n",
218 "spec = model_spec.get('mobilebert_classifier')"
227 …data specific to an on-device ML app and preprocess the data according to a specific `model_spec`."
242 " model_spec=spec,\n",
249 " model_spec=spec,\n",
271 "model = text_classifier.create(train_data, model_spec=spec)"
354 "## Choose a `model_spec` that Represents a Model for Text Classifier\n",
356 …"Each `model_spec` object represents a specific model for the text classifier. TensorFlow Lite Mod…
358 "Supported Model | Name of model_spec | Model Description\n",
375 "spec = model_spec.get('average_word_vec')"
[all …]
Dmodel_maker_question_answer.ipynb130 "spec = model_spec.get('mobilebert_qa')\n",
137 "model = question_answer.create(train_data, model_spec=spec)\n",
203 "from tflite_model_maker import model_spec\n",
223 "## Choose a model_spec that represents a model for question answer\n",
225 …"Each `model_spec` object represents a specific model for question answer. The Model Maker current…
227 "Supported Model | Name of model_spec | Model Description\n",
244 "spec = model_spec.get('mobilebert_qa_squad')"
299 …s://rajpurkar.github.io/SQuAD-explorer/) data according to a specific `model_spec`. You can use ei…
324 "1. Creates the model for question answer according to `model_spec`.\n",
325 …ng to two variables `default_training_epochs` and `default_batch_size` in the `model_spec` object."
[all …]
Dmodel_maker_image_classification.ipynb129 "from tflite_model_maker import model_spec\n",
734 …"We could switch model to MobileNetV2 by just setting parameter `model_spec` to `mobilenet_v2_spe…
745 …"model = image_classifier.create(train_data, model_spec=model_spec.mobilenet_v2_spec, validation_d…
791 "inception_v3_spec = model_spec.ImageModelSpec(\n",
802 …"Then, by setting parameter `model_spec` to `inception_v3_spec` in `create` method, we could retra…
/external/libtextclassifier/native/annotator/
Dmodel-executor.h35 const tflite::Model* model_spec) { in FromModelSpec() argument
36 auto model = TfLiteModelFromModelSpec(model_spec); in FromModelSpec()
/external/libtextclassifier/native/utils/
Dtflite-model-executor.h55 const tflite::Model* model_spec) { in FromModelSpec() argument
56 auto model = TfLiteModelFromModelSpec(model_spec); in FromModelSpec()
Dtflite-model-executor.cc308 const tflite::Model* model_spec) { in TfLiteModelFromModelSpec() argument
310 tflite::FlatBufferModel::BuildFromModel(model_spec)); in TfLiteModelFromModelSpec()