aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/Layer.hpp
diff options
context:
space:
mode:
authorFinn Williams <Finn.Williams@arm.com>2021-02-09 15:56:23 +0000
committerFinn Williams <Finn.Williams@arm.com>2021-02-12 13:10:20 +0000
commitb454c5c65efb238c130b042ace390b2bc7f0bf75 (patch)
treed6681d0abf416b3cc280bc3bb70e7d55dfd40a0d /src/armnn/Layer.hpp
parent8eae955f665f371b0a2c7c1a06e8ba442afa2298 (diff)
downloadarmnn-b454c5c65efb238c130b042ace390b2bc7f0bf75.tar.gz
IVGCVSW-4893 Refactor ILayerVisitor using unified interface strategy.
Signed-off-by: Jan Eilers <jan.eilers@arm.com> Signed-off-by: Finn Williams <Finn.Williams@arm.com> Signed-off-by: Francis Murtagh <francis.murtagh@arm.com> Change-Id: Id7bc8255a8e3f9e5aac65d510bec8a559bf37246
Diffstat (limited to 'src/armnn/Layer.hpp')
-rw-r--r--src/armnn/Layer.hpp10
1 files changed, 9 insertions, 1 deletions
diff --git a/src/armnn/Layer.hpp b/src/armnn/Layer.hpp
index ef0f8c3297..2f5cacc3ce 100644
--- a/src/armnn/Layer.hpp
+++ b/src/armnn/Layer.hpp
@@ -214,6 +214,9 @@ public:
Layer(unsigned int numInputSlots, unsigned int numOutputSlots, LayerType type, const char* name);
Layer(unsigned int numInputSlots, unsigned int numOutputSlots, LayerType type, DataLayout layout, const char* name);
+ void ExecuteStrategy(IStrategy& strategy) const override;
+
+
const std::string& GetNameStr() const
{
return m_LayerName;
@@ -259,7 +262,7 @@ public:
void ResetPriority() const;
LayerPriority GetPriority() const;
- LayerType GetType() const { return m_Type; }
+ LayerType GetType() const override { return m_Type; }
DataType GetDataType() const;
@@ -440,6 +443,11 @@ public:
LayerBindingId GetBindingId() const { return m_Id; };
+ void ExecuteStrategy(IStrategy& strategy) const override
+ {
+ strategy.ExecuteStrategy(this, BaseDescriptor(), {}, GetName(), GetBindingId());
+ }
+
protected:
~BindableLayer() = default;