diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2021-10-06 16:37:02 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-10-28 08:17:26 +0000 |
commit | b7851f9b29dbbe995acf6dc271471e73261c196a (patch) | |
tree | b00b29ea8e4161b11473053a0957f4a902aca837 /src/armnn/BackendHelper.cpp | |
parent | af3a4ef77d8f330a995911b979417857514df62c (diff) | |
download | armnn-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/armnn/BackendHelper.cpp')
-rw-r--r-- | src/armnn/BackendHelper.cpp | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/src/armnn/BackendHelper.cpp b/src/armnn/BackendHelper.cpp index cc792a06ef..c3cebddb2b 100644 --- a/src/armnn/BackendHelper.cpp +++ b/src/armnn/BackendHelper.cpp @@ -126,6 +126,18 @@ bool IsCapabilitySupported(const armnn::BackendId& backend, armnn::BackendCapabi return hasCapability; } +unsigned int GetNumberOfCacheFiles(const armnn::BackendId& backend) +{ + auto const& backendRegistry = armnn::BackendRegistryInstance(); + if (backendRegistry.IsBackendRegistered(backend)) + { + auto factoryFunc = backendRegistry.GetFactory(backend); + auto backendObject = factoryFunc(); + return backendObject->GetNumberOfCacheFiles(); + } + return 0; +} + bool LayerSupportHandle::IsBackendRegistered() const { if (m_LayerSupport) |