diff options
Diffstat (limited to 'source/application/api/use_case/inference_runner/src/TestModel.cc')
-rw-r--r-- | source/application/api/use_case/inference_runner/src/TestModel.cc | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/source/application/api/use_case/inference_runner/src/TestModel.cc b/source/application/api/use_case/inference_runner/src/TestModel.cc index c69a98d..94f17ef 100644 --- a/source/application/api/use_case/inference_runner/src/TestModel.cc +++ b/source/application/api/use_case/inference_runner/src/TestModel.cc @@ -16,8 +16,15 @@ */ #include "TestModel.hpp" #include "log_macros.h" +#include "MicroMutableAllOpsResolver.hpp" -const tflite::MicroMutableOpResolver<kNumberOperators>& arm::app::TestModel::GetOpResolver() +const tflite::MicroOpResolver& arm::app::TestModel::GetOpResolver() { - return this->m_opResolver; + return this->m_opResolver; +} + +bool arm::app::TestModel::EnlistOperations() +{ + this->m_opResolver = CreateAllOpsResolver(); + return true; } |