aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl/ClBackendModelContext.cpp
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2021-10-06 16:37:02 +0100
committerSadik Armagan <sadik.armagan@arm.com>2021-10-28 08:17:26 +0000
commitb7851f9b29dbbe995acf6dc271471e73261c196a (patch)
treeb00b29ea8e4161b11473053a0957f4a902aca837 /src/backends/cl/ClBackendModelContext.cpp
parentaf3a4ef77d8f330a995911b979417857514df62c (diff)
downloadarmnn-b7851f9b29dbbe995acf6dc271471e73261c196a.tar.gz
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 <sadik.armagan@arm.com> Change-Id: Ie02ac123bb7df9b0593a2fe46b5bb564a5994780
Diffstat (limited to 'src/backends/cl/ClBackendModelContext.cpp')
-rw-r--r--src/backends/cl/ClBackendModelContext.cpp11
1 files changed, 10 insertions, 1 deletions
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