From b7851f9b29dbbe995acf6dc271471e73261c196a Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Wed, 6 Oct 2021 16:37:02 +0100 Subject: 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 Change-Id: Ie02ac123bb7df9b0593a2fe46b5bb564a5994780 --- src/armnn/OptimizedNetworkImpl.hpp | 3 +++ 1 file changed, 3 insertions(+) (limited to 'src/armnn/OptimizedNetworkImpl.hpp') 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; } -- cgit v1.2.1