diff options
| author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2023-05-05 16:39:05 +0100 |
|---|---|---|
| committer | Francis Murtagh <francis.murtagh@arm.com> | 2023-05-08 13:19:15 +0000 |
| commit | 46e574e8195ae2e8085086457a642210d9d3b8cd (patch) | |
| tree | b3a7b2bdc86b222509f5110c177a82646a58797b /tests/ExecuteNetwork/ExecuteNetwork.cpp | |
| parent | 97a3aefff63ae081ae62aa5bac17d6e9c401937e (diff) | |
| download | armnn-46e574e8195ae2e8085086457a642210d9d3b8cd.tar.gz | |
IVGCVSW-7626 Add Execute Network for Opaque Delegate
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ibdded86713368ecfdf31c4118dfe8a3404d1e3b8
Diffstat (limited to 'tests/ExecuteNetwork/ExecuteNetwork.cpp')
| -rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index 14841ec1e9..f9f583a9c6 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -5,7 +5,7 @@ #include "ExecuteNetworkProgramOptions.hpp" #include "ArmNNExecutor.hpp" -#if defined(ARMNN_TFLITE_DELEGATE) +#if defined(ARMNN_TFLITE_DELEGATE) || defined(ARMNN_TFLITE_OPAQUE_DELEGATE) #include "TfliteExecutor.hpp" #endif #include <armnn/Logging.hpp> @@ -13,10 +13,12 @@ std::unique_ptr<IExecutor> BuildExecutor(ProgramOptions& programOptions) { - if (programOptions.m_ExNetParams.m_TfLiteExecutor == ExecuteNetworkParams::TfLiteExecutor::ArmNNTfLiteDelegate || + if (programOptions.m_ExNetParams.m_TfLiteExecutor == + ExecuteNetworkParams::TfLiteExecutor::ArmNNTfLiteOpaqueDelegate || + programOptions.m_ExNetParams.m_TfLiteExecutor == ExecuteNetworkParams::TfLiteExecutor::ArmNNTfLiteDelegate || programOptions.m_ExNetParams.m_TfLiteExecutor == ExecuteNetworkParams::TfLiteExecutor::TfliteInterpreter) { -#if defined(ARMNN_TFLITE_DELEGATE) +#if defined(ARMNN_TFLITE_DELEGATE) || defined(ARMNN_TFLITE_OPAQUE_DELEGATE) return std::make_unique<TfLiteExecutor>(programOptions.m_ExNetParams, programOptions.m_RuntimeOptions); #else ARMNN_LOG(fatal) << "Not built with Arm NN Tensorflow-Lite delegate support."; |