diff options
Diffstat (limited to 'tests/ExecuteNetwork')
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index abcda40d1d..1f19584c68 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -11,6 +11,7 @@ #include <armnn/Logging.hpp> #include <armnnUtils/Filesystem.hpp> #include <armnnUtils/TContainer.hpp> +#include <ProfilingOptionsConverter.hpp> #include <InferenceTest.hpp> #if defined(ARMNN_SERIALIZER) @@ -90,7 +91,8 @@ int TfLiteDelegateMainImpl(const ExecuteNetworkParams& params, const armnn::IRun // Create the Armnn Delegate // Populate a DelegateOptions from the ExecuteNetworkParams. armnnDelegate::DelegateOptions delegateOptions = params.ToDelegateOptions(); - delegateOptions.SetExternalProfilingParams(runtimeOptions.m_ProfilingOptions); + delegateOptions.SetExternalProfilingParams( + ConvertExternalProfilingOptions(runtimeOptions.m_ProfilingOptions)); std::unique_ptr<TfLiteDelegate, decltype(&armnnDelegate::TfLiteArmnnDelegateDelete)> theArmnnDelegate(armnnDelegate::TfLiteArmnnDelegateCreate(delegateOptions), |