From 615e06f54a4c4139e81e289991ba4084aa2f69d3 Mon Sep 17 00:00:00 2001 From: Finn Williams Date: Mon, 20 Jun 2022 13:48:20 +0100 Subject: IVGCVSW-6650 Refactor ExecuteNetwork * Remove InferenceModel * Add automatic IO type, shape and name configuration * Depreciate various redundant options * Add internal output comparison Signed-off-by: Finn Williams Change-Id: I2eca248bc91e1655a99ed94990efb8059f541fa9 --- tests/CMakeLists.txt | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'tests/CMakeLists.txt') diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt index 4cb324f2c7..87a5b46024 100644 --- a/tests/CMakeLists.txt +++ b/tests/CMakeLists.txt @@ -139,6 +139,9 @@ if (BUILD_ARMNN_SERIALIZER OR BUILD_ONNX_PARSER OR BUILD_ARMNN_TFLITE_DELEGATE) set(ExecuteNetwork_sources + ExecuteNetwork/IExecutor.hpp + ExecuteNetwork/ArmNNExecutor.cpp + ExecuteNetwork/ArmNNExecutor.hpp ExecuteNetwork/ExecuteNetwork.cpp ExecuteNetwork/ExecuteNetworkProgramOptions.cpp ExecuteNetwork/ExecuteNetworkProgramOptions.hpp @@ -147,6 +150,13 @@ if (BUILD_ARMNN_SERIALIZER NetworkExecutionUtils/NetworkExecutionUtils.cpp NetworkExecutionUtils/NetworkExecutionUtils.hpp) + if(BUILD_ARMNN_TFLITE_DELEGATE) + set(ExecuteNetwork_sources + ${ExecuteNetwork_sources} + ExecuteNetwork/TfliteExecutor.cpp + ExecuteNetwork/TfliteExecutor.hpp) + endif() + add_executable_ex(ExecuteNetwork ${ExecuteNetwork_sources}) target_include_directories(ExecuteNetwork PRIVATE ../src/armnn) target_include_directories(ExecuteNetwork PRIVATE ../src/armnnUtils) -- cgit v1.2.1