From 2cd184763ff7f8767e751f2fe0c461714350aae6 Mon Sep 17 00:00:00 2001 From: Jan Eilers Date: Tue, 15 Dec 2020 10:42:38 +0000 Subject: IVGCVSW-5571 Expose the TfLite Delegate to the TfLite python API * Implemented external delegate adaptor interface for TfLite * Activated armnn logging for delegate * Added logging info to indicate if gpu tuning is turned on * Added pytests to ensure functionality of the external delegate adaptor * Included the delegate directory into doxygen * Added documentation on how to use the external delegate in python Signed-off-by: Finn Williams Signed-off-by: Jan Eilers Change-Id: Id3b4588fb0b9ac7e3f47ba2c19feead7beb58e18 --- include/armnn/Logging.hpp | 39 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 39 insertions(+) (limited to 'include/armnn') diff --git a/include/armnn/Logging.hpp b/include/armnn/Logging.hpp index 27971c56e4..9a60e07c63 100644 --- a/include/armnn/Logging.hpp +++ b/include/armnn/Logging.hpp @@ -7,6 +7,7 @@ #include #include +#include namespace armnn { @@ -32,6 +33,44 @@ inline std::string LevelToString(LogSeverity level) } } +inline LogSeverity StringToLogLevel(std::string level) +{ + // Transfer to lower case + std::transform(level.begin(), level.end(), level.begin(), + [](unsigned char c){ return std::tolower(c); } + ); + + if (level == "trace") + { + return LogSeverity::Trace; + } + else if (level == "debug") + { + return LogSeverity::Debug; + } + else if (level == "info") + { + return LogSeverity::Info; + } + else if (level == "warning") + { + return LogSeverity::Warning; + } + else if (level == "error") + { + return LogSeverity::Error; + } + else if (level == "fatal") + { + return LogSeverity::Fatal; + } + else + { + throw armnn::Exception("Unknown severity level for logging: '" + level + + "'. Valid options: trace, debug, info, warning, error, fatal"); + } +} + class LogSink { public: -- cgit v1.2.1