commit b7851f9b29dbbe995acf6dc271471e73261c196a
author Sadik Armagan <sadik.armagan@arm.com> Wed Oct 06 16:37:02 2021 +0100
committer Sadik Armagan <sadik.armagan@arm.com> Thu Oct 28 08:17:26 2021 +0000
tree b00b29ea8e4161b11473053a0957f4a902aca837
parent af3a4ef77d8f330a995911b979417857514df62c

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

src/armnn/OptimizedNetworkImpl.hpp

diff --git a/src/armnn/OptimizedNetworkImpl.hpp b/src/armnn/OptimizedNetworkImpl.hpp
index d42cff7..112d585 100644
--- a/src/armnn/OptimizedNetworkImpl.hpp
+++ b/src/armnn/OptimizedNetworkImpl.hpp
@@ -21,6 +21,9 @@
 
     virtual profiling::ProfilingGuid GetGuid() const { return m_Guid; };
 
+    virtual size_t GetNumInputs() const;
+    virtual size_t GetNumOutputs() const;
+
     Graph& GetGraph() { return *m_Graph; }
     ModelOptions& GetModelOptions() { return m_ModelOptions; }