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/Layer.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/Layer.hpp')
-rw-r--r-- | src/armnn/Layer.hpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/armnn/Layer.hpp b/src/armnn/Layer.hpp index aab5227b75..ad6c9b298e 100644 --- a/src/armnn/Layer.hpp +++ b/src/armnn/Layer.hpp @@ -406,7 +406,11 @@ protected: // Retrieve the Handles to the constants // Marking this as override and having this here keeps IConnectable abstract with only pure virtual function - virtual ConstantTensors GetConstantTensorsByRef() override {return ConstantTensors(); }; + virtual ConstantTensors GetConstantTensorsByRef() override final; + + // Retrieve the Handles to the constants + // Marking this as override and having this here keeps IConnectable abstract with only pure virtual function + virtual ImmutableConstantTensors GetConstantTensorsByRef() const override { return ImmutableConstantTensors(); }; // "Blob" AdditionalInfoObjectPtr m_AdditionalInfoObject; |