Searched refs:TfLiteModelExecutor (Results 1 – 10 of 10) sorted by relevance
| /external/libtextclassifier/native/utils/ |
| D | tflite-model-executor.h | 52 class TfLiteModelExecutor {
54 static std::unique_ptr<TfLiteModelExecutor> FromModelSpec( in FromModelSpec()
60 return std::unique_ptr<TfLiteModelExecutor>( in FromModelSpec()
61 new TfLiteModelExecutor(std::move(model))); in FromModelSpec()
64 static std::unique_ptr<TfLiteModelExecutor> FromBuffer( in FromBuffer()
70 return std::unique_ptr<TfLiteModelExecutor>( in FromBuffer()
71 new TfLiteModelExecutor(std::move(model))); in FromBuffer()
144 explicit TfLiteModelExecutor(
146 TfLiteModelExecutor(std::unique_ptr<const tflite::FlatBufferModel> model,
154 void TfLiteModelExecutor::SetInput(const int input_index,
[all …]
|
| D | tflite-model-executor.cc | 330 TfLiteModelExecutor::TfLiteModelExecutor( in TfLiteModelExecutor() function in libtextclassifier3::TfLiteModelExecutor
333 TfLiteModelExecutor::TfLiteModelExecutor( in TfLiteModelExecutor() function in libtextclassifier3::TfLiteModelExecutor
338 std::unique_ptr<tflite::Interpreter> TfLiteModelExecutor::CreateInterpreter() in CreateInterpreter()
346 void TfLiteModelExecutor::SetInput(const int input_index, in SetInput()
358 std::vector<tflite::StringRef> TfLiteModelExecutor::Output( in Output()
371 std::vector<std::string> TfLiteModelExecutor::Output( in Output()
|
| /external/libtextclassifier/native/annotator/ |
| D | model-executor.h | 32 class ModelExecutor : public TfLiteModelExecutor {
57 : TfLiteModelExecutor(std::move(model)) {} in ModelExecutor()
98 std::unique_ptr<TfLiteModelExecutor> executor, int quantization_bits,
104 std::unique_ptr<TfLiteModelExecutor> executor_;
|
| D | model-executor.cc | 49 std::unique_ptr<TfLiteModelExecutor> executor = in FromBuffer()
50 TfLiteModelExecutor::FromBuffer(model_spec_buffer); in FromBuffer()
90 std::unique_ptr<TfLiteModelExecutor> executor, int quantization_bits, in TFLiteEmbeddingExecutor()
|
| /external/libtextclassifier/native/actions/ |
| D | lua-actions.cc | 35 const TfLiteModelExecutor* model_executor, in GetTensorViewForOutput()
44 const TfLiteModelExecutor* model_executor, in GetStringTensorForOutput()
57 const TfLiteModelExecutor* model_executor, in CreateLuaActionsSuggestions()
76 const TfLiteModelExecutor* model_executor, in LuaActionsSuggestions()
|
| D | lua-actions.h | 33 const TfLiteModelExecutor* model_executor,
44 const TfLiteModelExecutor* model_executor,
|
| D | tflite-sensitive-model.h | 41 std::unique_ptr<const TfLiteModelExecutor> model_executor_;
|
| D | actions-suggestions.h | 225 const TfLiteModelExecutor* model_executor,
238 std::unique_ptr<const TfLiteModelExecutor> model_executor_;
|
| D | tflite-sensitive-model.cc | 126 model_executor_(TfLiteModelExecutor::FromBuffer( in TFLiteSensitiveModel()
|
| D | actions-suggestions.cc | 99 const std::unique_ptr<const TfLiteModelExecutor>& model_executor) { in SetVectorOrScalarAsModelInput()
279 model_executor_ = TfLiteModelExecutor::FromBuffer( in ValidateAndInitialize()
1272 const Conversation& conversation, const TfLiteModelExecutor* model_executor, in SuggestActionsFromLua()
1291 const Conversation& conversation, const TfLiteModelExecutor* model_executor, in SuggestActionsFromLua()
|