From 2e24175c683bca42496104591d6b702dad360b8e Mon Sep 17 00:00:00 2001
From: Nikhil Raj <nikhil.raj@arm.com>
Date: Tue, 1 Feb 2022 16:42:15 +0000
Subject: IVGCVSW-6724 Accessing ConstTensors from IConnectableLayer

Signed-off-by: Nikhil Raj <nikhil.raj@arm.com>
Change-Id: I01f42a520d15c6dabd2f77c7715c91b8f7026476
---
 include/armnn/INetwork.hpp | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'include/armnn/INetwork.hpp')

diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp
index 073f119ef4..6a2193cbc6 100644
--- a/include/armnn/INetwork.hpp
+++ b/include/armnn/INetwork.hpp
@@ -13,6 +13,7 @@
 #include <armnn/Optional.hpp>
 #include <armnn/TensorFwd.hpp>
 #include <armnn/Logging.hpp>
+#include <armnn/backends/TensorHandle.hpp>
 
 #include <memory>
 #include <vector>
@@ -119,6 +120,11 @@ public:
     /// the BaseDescriptor IsNull function is invoked.
     virtual const BaseDescriptor& GetParameters() const = 0;
 
+    using ConstantTensors = std::vector<std::reference_wrapper<std::shared_ptr<ConstTensorHandle>>>;
+
+    // Returns ConstantTensors of this Layer if it has any, otherwise returns empty vector.
+    virtual ConstantTensors GetConstantTensorsByRef() = 0;
+
 protected:
       /// Objects are not deletable via the handle
     ~IConnectableLayer() {}
-- 
cgit v1.2.1