diff options
Diffstat (limited to 'delegate')
-rw-r--r-- | delegate/include/DelegateOptions.hpp | 7 | ||||
-rw-r--r-- | delegate/src/DelegateOptions.cpp | 2 |
2 files changed, 5 insertions, 4 deletions
diff --git a/delegate/include/DelegateOptions.hpp b/delegate/include/DelegateOptions.hpp index 5bc2e59070..5aa7939e4b 100644 --- a/delegate/include/DelegateOptions.hpp +++ b/delegate/include/DelegateOptions.hpp @@ -8,6 +8,7 @@ #include <armnn/ArmNN.hpp> #include <armnn/Logging.hpp> #include <armnn/Optional.hpp> +#include <armnn/profiling/ProfilingOptions.hpp> #include <set> #include <string> @@ -229,10 +230,10 @@ public: const armnn::ProfilingDetailsMethod& GetInternalProfilingDetail() const { return m_InternalProfilingDetail; } void SetExternalProfilingParams( - const armnn::IRuntime::CreationOptions::ExternalProfilingOptions& externalProfilingParams) + const armnn::profiling::ProfilingOptions& externalProfilingParams) { m_ProfilingOptions = externalProfilingParams; } - const armnn::IRuntime::CreationOptions::ExternalProfilingOptions& GetExternalProfilingParams() const + const armnn::profiling::ProfilingOptions& GetExternalProfilingParams() const { return m_ProfilingOptions; } void SetSerializeToDot(const std::string& serializeToDotFile) { m_SerializeToDot = serializeToDotFile; } @@ -264,7 +265,7 @@ private: armnn::OptimizerOptions m_OptimizerOptions; /// External profiling options. - armnn::IRuntime::CreationOptions::ExternalProfilingOptions m_ProfilingOptions; + armnn::profiling::ProfilingOptions m_ProfilingOptions; /// Internal profiling options. /// Indicates whether internal profiling is enabled or not. diff --git a/delegate/src/DelegateOptions.cpp b/delegate/src/DelegateOptions.cpp index d477d9839d..5cfddadf99 100644 --- a/delegate/src/DelegateOptions.cpp +++ b/delegate/src/DelegateOptions.cpp @@ -59,7 +59,7 @@ DelegateOptions::DelegateOptions(char const* const* options_keys, armnn::OptimizerOptions optimizerOptions; bool internalProfilingState = false; armnn::ProfilingDetailsMethod internalProfilingDetail = armnn::ProfilingDetailsMethod::DetailsWithEvents; - armnn::IRuntime::CreationOptions::ExternalProfilingOptions extProfilingParams; + armnn::profiling::ProfilingOptions extProfilingParams; for (size_t i = 0; i < num_options; ++i) { // Process backends |