Home
last modified time | relevance | path

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

/external/libtextclassifier/native/annotator/
Dmodel-executor.h32 class ModelExecutor : public TfLiteModelExecutor {
34 static std::unique_ptr<ModelExecutor> FromModelSpec( in FromModelSpec()
40 return std::unique_ptr<ModelExecutor>(new ModelExecutor(std::move(model))); in FromModelSpec()
43 static std::unique_ptr<ModelExecutor> FromBuffer( in FromBuffer()
49 return std::unique_ptr<ModelExecutor>(new ModelExecutor(std::move(model))); in FromBuffer()
56 explicit ModelExecutor(std::unique_ptr<const tflite::FlatBufferModel> model) in ModelExecutor() function
Dannotator.h69 InterpreterManager(const ModelExecutor* selection_executor, in InterpreterManager()
70 const ModelExecutor* classification_executor) in InterpreterManager()
81 const ModelExecutor* selection_executor_;
82 const ModelExecutor* classification_executor_;
445 std::unique_ptr<const ModelExecutor> selection_executor_;
446 std::unique_ptr<const ModelExecutor> classification_executor_;
Dmodel-executor.cc24 TensorView<float> ModelExecutor::ComputeLogits( in ComputeLogits()
Dannotator.cc336 selection_executor_ = ModelExecutor::FromBuffer(model_->selection_model()); in ValidateAndInitialize()
371 ModelExecutor::FromBuffer(model_->classification_model()); in ValidateAndInitialize()