aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl/ClBackend.hpp
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/ClBackend.hpp
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/ClBackend.hpp')
-rw-r--r--src/backends/cl/ClBackend.hpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/backends/cl/ClBackend.hpp b/src/backends/cl/ClBackend.hpp
index e0708d18e2..ffce800261 100644
--- a/src/backends/cl/ClBackend.hpp
+++ b/src/backends/cl/ClBackend.hpp
@@ -105,6 +105,8 @@ public:
return m_UsingCustomAllocator;
}
+ virtual unsigned int GetNumberOfCacheFiles() const override { return 1; }
+
// Cl requires a arm_compute::IAllocator we wrap the Arm NN ICustomAllocator to achieve this
class ClBackendCustomAllocatorWrapper : public arm_compute::IAllocator
{