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 /src/armnn/layers/LayerWithParameters.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 'src/armnn/layers/LayerWithParameters.hpp')
-rw-r--r-- | src/armnn/layers/LayerWithParameters.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnn/layers/LayerWithParameters.hpp b/src/armnn/layers/LayerWithParameters.hpp index 40ade95d5c..0a1dbf363c 100644 --- a/src/armnn/layers/LayerWithParameters.hpp +++ b/src/armnn/layers/LayerWithParameters.hpp @@ -56,9 +56,9 @@ protected: strategy.ExecuteStrategy(this, GetParameters(), {}, GetName()); } - Layer::ConstantTensors GetConnectedConstantAsInputTensors() + Layer::ImmutableConstantTensors GetConnectedConstantAsInputTensors() const { - Layer::ConstantTensors tensors; + Layer::ImmutableConstantTensors tensors; for (unsigned int i = 0; i < GetNumInputSlots(); ++i) { if (GetInputSlot(i).GetConnection() && GetInputSlot(i).GetConnection()->GetTensorInfo().IsConstant()) |