diff options
author | Nikhil Raj <nikhil.raj@arm.com> | 2022-02-01 16:42:15 +0000 |
---|---|---|
committer | Nikhil Raj Arm <nikhil.raj@arm.com> | 2022-02-03 14:36:14 +0000 |
commit | 2e24175c683bca42496104591d6b702dad360b8e (patch) | |
tree | 1802062ab9132e3196e76b083b1a7b2b6c6efd11 /include/armnn | |
parent | ab8a4465ba32cb00ad05d63abfd2e60c307edc51 (diff) | |
download | armnn-2e24175c683bca42496104591d6b702dad360b8e.tar.gz |
IVGCVSW-6724 Accessing ConstTensors from IConnectableLayer
Signed-off-by: Nikhil Raj <nikhil.raj@arm.com>
Change-Id: I01f42a520d15c6dabd2f77c7715c91b8f7026476
Diffstat (limited to 'include/armnn')
-rw-r--r-- | include/armnn/INetwork.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
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() {} |