diff options
author | Matthew Bentham <matthew.bentham@arm.com> | 2023-02-23 13:03:46 +0000 |
---|---|---|
committer | Francis Murtagh <francis.murtagh@arm.com> | 2023-02-28 12:36:12 +0000 |
commit | aeec3ce5c8f936fb1220a9de8c84cceef88d4080 (patch) | |
tree | 87e676cecdd4f81be53e33c484721f9448e313a2 /include/armnn/INetwork.hpp | |
parent | 2a764ade6b5bf88cba0c43303291e0352ec3354c (diff) | |
download | armnn-aeec3ce5c8f936fb1220a9de8c84cceef88d4080.tar.gz |
Add constant version of IConnectableLayer::GetConstantTensorsByRef
This makes it easier to use, particularly in backends where it
is common to pass around const pointers to IConnectableLayer.
The non-constant version is rewritten to use the constant version.
Signed-off-by: Matthew Bentham <matthew.bentham@arm.com>
Change-Id: Id3a8384447e93c213299a85ade9a667df5960534
Diffstat (limited to 'include/armnn/INetwork.hpp')
-rw-r--r-- | include/armnn/INetwork.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp index c9c8a04656..c944d09d41 100644 --- a/include/armnn/INetwork.hpp +++ b/include/armnn/INetwork.hpp @@ -126,6 +126,11 @@ public: // Returns ConstantTensors of this Layer if it has any, otherwise returns empty vector. virtual ConstantTensors GetConstantTensorsByRef() = 0; + using ImmutableConstantTensors = std::vector<std::reference_wrapper<const std::shared_ptr<ConstTensorHandle>>>; + + // Returns ConstantTensors of this Layer if it has any, otherwise returns empty vector. + virtual ImmutableConstantTensors GetConstantTensorsByRef() const = 0; + protected: /// Objects are not deletable via the handle ~IConnectableLayer() {} |