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/OptimizedNetworkImpl.hpp | |
| 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/OptimizedNetworkImpl.hpp')
| -rw-r--r-- | src/armnn/OptimizedNetworkImpl.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnn/OptimizedNetworkImpl.hpp b/src/armnn/OptimizedNetworkImpl.hpp index d42cff7346..112d585aee 100644 --- a/src/armnn/OptimizedNetworkImpl.hpp +++ b/src/armnn/OptimizedNetworkImpl.hpp @@ -21,6 +21,9 @@ public: 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; } |