Searched refs:TfLiteModelExecutor (Results 1 – 8 of 8) sorted by relevance
| /external/libtextclassifier/utils/ |
| D | tflite-model-executor.h | 41 class TfLiteModelExecutor {
43 static std::unique_ptr<TfLiteModelExecutor> FromModelSpec( in FromModelSpec()
49 return std::unique_ptr<TfLiteModelExecutor>( in FromModelSpec()
50 new TfLiteModelExecutor(std::move(model))); in FromModelSpec()
53 static std::unique_ptr<TfLiteModelExecutor> FromBuffer( in FromBuffer()
59 return std::unique_ptr<TfLiteModelExecutor>( in FromBuffer()
60 new TfLiteModelExecutor(std::move(model))); in FromBuffer()
133 explicit TfLiteModelExecutor(
141 void TfLiteModelExecutor::SetInput(const int input_index,
146 std::vector<tflite::StringRef> TfLiteModelExecutor::Output(
[all …]
|
| D | tflite-model-executor.cc | 201 TfLiteModelExecutor::TfLiteModelExecutor( in TfLiteModelExecutor() function in libtextclassifier3::TfLiteModelExecutor
205 std::unique_ptr<tflite::Interpreter> TfLiteModelExecutor::CreateInterpreter() in CreateInterpreter()
213 void TfLiteModelExecutor::SetInput(const int input_index, in SetInput()
225 std::vector<tflite::StringRef> TfLiteModelExecutor::Output( in Output()
238 std::vector<std::string> TfLiteModelExecutor::Output( in Output()
|
| /external/libtextclassifier/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/actions/ |
| D | lua-actions.cc | 33 const TfLiteModelExecutor* model_executor, in GetTensorViewForOutput()
52 const TfLiteModelExecutor* model_executor, in CreateLuaActionsSuggestions()
71 const TfLiteModelExecutor* model_executor, in LuaActionsSuggestions()
|
| D | lua-actions.h | 34 const TfLiteModelExecutor* model_executor,
54 const TfLiteModelExecutor* model_executor,
|
| D | actions-suggestions.h | 233 const TfLiteModelExecutor* model_executor,
266 std::unique_ptr<const TfLiteModelExecutor> model_executor_;
|
| D | actions-suggestions.cc | 251 model_executor_ = TfLiteModelExecutor::FromBuffer( in ValidateAndInitialize()
1302 const Conversation& conversation, const TfLiteModelExecutor* model_executor, in SuggestActionsFromLua()
|