summaryrefslogtreecommitdiff
path: root/source/application/api/use_case/inference_runner/src/TestModel.cc
diff options
context:
space:
mode:
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.cc11
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;
}