summaryrefslogtreecommitdiff
path: root/source/application/api/use_case/inference_runner/include/TestModel.hpp
diff options
context:
space:
mode:
authorNina Drozd <nina.drozd@arm.com>2023-08-22 12:41:55 +0100
committerNina Drozd <nina.drozd@arm.com>2023-08-31 08:41:00 +0000
commit05c1c5a1beb8c1c17e2a06000727f9f094bc6e37 (patch)
treec31ae9200817fa9c7c5828bec1a06c623d5251df /source/application/api/use_case/inference_runner/include/TestModel.hpp
parent677d43fa8f55a8aa52e6bd9d1884e2797650fd65 (diff)
downloadml-embedded-evaluation-kit-05c1c5a1beb8c1c17e2a06000727f9f094bc6e37.tar.gz
MLECO-4072: update MLEK dependencies to 23.08
* version updates in cmakelists file * update submodules to 23.08 versions * updates to documentation - vela 3.9.0 Change-Id: I5e24d5729ecbd2289b04fcc845fb44f455ab38d5 Signed-off-by: Nina Drozd <nina.drozd@arm.com>
Diffstat (limited to 'source/application/api/use_case/inference_runner/include/TestModel.hpp')
-rw-r--r--source/application/api/use_case/inference_runner/include/TestModel.hpp7
1 files changed, 4 insertions, 3 deletions
diff --git a/source/application/api/use_case/inference_runner/include/TestModel.hpp b/source/application/api/use_case/inference_runner/include/TestModel.hpp
index 80318e6..bc6a0fb 100644
--- a/source/application/api/use_case/inference_runner/include/TestModel.hpp
+++ b/source/application/api/use_case/inference_runner/include/TestModel.hpp
@@ -1,5 +1,5 @@
/*
- * SPDX-FileCopyrightText: Copyright 2021 Arm Limited and/or its affiliates <open-source-office@arm.com>
+ * SPDX-FileCopyrightText: Copyright 2021, 2023 Arm Limited and/or its affiliates <open-source-office@arm.com>
* SPDX-License-Identifier: Apache-2.0
*
* Licensed under the Apache License, Version 2.0 (the "License");
@@ -18,6 +18,7 @@
#define INF_RUNNER_TESTMODEL_HPP
#include "Model.hpp"
+#include "MicroMutableAllOpsResolver.hpp"
namespace arm {
namespace app {
@@ -26,7 +27,7 @@ namespace app {
protected:
/** @brief Gets the reference to op resolver interface class. */
- const tflite::AllOpsResolver& GetOpResolver() override;
+ const tflite::MicroMutableOpResolver<kNumberOperators>& GetOpResolver() override;
/** @brief Adds operations to the op resolver instance, not needed as using AllOpsResolver. */
bool EnlistOperations() override {return false;}
@@ -34,7 +35,7 @@ namespace app {
private:
/* No need to define individual ops at the cost of extra memory. */
- tflite::AllOpsResolver m_opResolver;
+ tflite::MicroMutableOpResolver<kNumberOperators> m_opResolver;
};
} /* namespace app */