diff options
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; } |