From b7851f9b29dbbe995acf6dc271471e73261c196a Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Wed, 6 Oct 2021 16:37:02 +0100 Subject: IVGCVSW-5636 'Implement NNAPI caching functions' * Get number of inputs and outputs from optimized network. * Get number of cached files if backend supports caching. Signed-off-by: Sadik Armagan Change-Id: Ie02ac123bb7df9b0593a2fe46b5bb564a5994780 --- src/backends/cl/ClBackendModelContext.cpp | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) (limited to 'src/backends/cl/ClBackendModelContext.cpp') diff --git a/src/backends/cl/ClBackendModelContext.cpp b/src/backends/cl/ClBackendModelContext.cpp index b685bc296c..75a2e05bda 100644 --- a/src/backends/cl/ClBackendModelContext.cpp +++ b/src/backends/cl/ClBackendModelContext.cpp @@ -32,7 +32,7 @@ namespace armnn { ClBackendModelContext::ClBackendModelContext(const ModelOptions& modelOptions) - : m_CachedNetworkFilePath(""), m_IsFastMathEnabled(false), m_SaveCachedNetwork(false) + : m_CachedNetworkFilePath(""), m_IsFastMathEnabled(false), m_SaveCachedNetwork(false), m_CachedFileDescriptor(-1) { if (!modelOptions.empty()) { @@ -50,6 +50,10 @@ ClBackendModelContext::ClBackendModelContext(const ModelOptions& modelOptions) { m_CachedNetworkFilePath = ParseFile(value, ""); } + if (name == "CachedFileDescriptor") + { + m_CachedFileDescriptor = armnn::ParseIntBackendOption(value, -1); + } }); } } @@ -69,4 +73,9 @@ bool ClBackendModelContext::SaveCachedNetwork() const return m_SaveCachedNetwork; } +int ClBackendModelContext::GetCachedFileDescriptor() const +{ + return m_CachedFileDescriptor; +} + } // namespace armnn \ No newline at end of file -- cgit v1.2.1