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/Network.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/Network.cpp')
| -rw-r--r-- | src/armnn/Network.cpp | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index b516d519d5..e00dbfc0fc 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -524,6 +524,16 @@ profiling::ProfilingGuid IOptimizedNetwork::GetGuid() const return pOptimizedNetworkImpl->GetGuid(); } +size_t IOptimizedNetwork::GetNumInputs() const +{ + return pOptimizedNetworkImpl->GetNumInputs(); +} + +size_t IOptimizedNetwork::GetNumOutputs() const +{ + return pOptimizedNetworkImpl->GetNumOutputs(); +} + Status OptimizedNetworkImpl::PrintGraph() { m_Graph->Print(); @@ -535,6 +545,16 @@ Status OptimizedNetworkImpl::SerializeToDot(std::ostream& stream) const return m_Graph->SerializeToDot(stream); } +size_t OptimizedNetworkImpl::GetNumInputs() const +{ + return m_Graph->GetNumInputs(); +} + +size_t OptimizedNetworkImpl::GetNumOutputs() const +{ + return m_Graph->GetNumOutputs(); +} + void ReportError(const std::string& errorMessage, Optional<std::vector<std::string>&> errorMessages) { |