aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/OptimizedNetworkImpl.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/OptimizedNetworkImpl.hpp')
-rw-r--r--src/armnn/OptimizedNetworkImpl.hpp3
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; }