diff options
| author | Finn Williams <Finn.Williams@arm.com> | 2021-02-09 15:56:23 +0000 |
|---|---|---|
| committer | Finn Williams <Finn.Williams@arm.com> | 2021-02-12 13:10:20 +0000 |
| commit | b454c5c65efb238c130b042ace390b2bc7f0bf75 (patch) | |
| tree | d6681d0abf416b3cc280bc3bb70e7d55dfd40a0d /include/armnn/INetwork.hpp | |
| parent | 8eae955f665f371b0a2c7c1a06e8ba442afa2298 (diff) | |
| download | armnn-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 'include/armnn/INetwork.hpp')
| -rw-r--r-- | include/armnn/INetwork.hpp | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp index ca1b725d48..c667d9ce8b 100644 --- a/include/armnn/INetwork.hpp +++ b/include/armnn/INetwork.hpp @@ -8,6 +8,7 @@ #include <armnn/Deprecated.hpp> #include <armnn/DescriptorsFwd.hpp> #include <armnn/ILayerVisitor.hpp> +#include <armnn/IStrategy.hpp> #include <armnn/NetworkFwd.hpp> #include <armnn/Optional.hpp> #include <armnn/TensorFwd.hpp> @@ -91,8 +92,15 @@ public: /// Apply a visitor to this layer virtual void Accept(ILayerVisitor& visitor) const = 0; + /// Apply a visitor to this layer + virtual void ExecuteStrategy(IStrategy& strategy) const = 0; + /// Provide a hint for the optimizer as to which backend to prefer for this layer virtual void BackendSelectionHint(Optional<BackendId> backend) = 0; + + /// Returns the armnn::LayerType of this layer + virtual LayerType GetType() const = 0; + protected: /// Objects are not deletable via the handle ~IConnectableLayer() {} @@ -600,6 +608,8 @@ public: virtual void Accept(ILayerVisitor& visitor) const = 0; + virtual void ExecuteStrategy(IStrategy& strategy) const = 0; + protected: ~INetwork() {} }; |