diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2023-08-17 18:44:58 +0100 |
|---|---|---|
| committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2023-08-28 12:37:25 +0100 |
| commit | 9145e38edf49fa4862008c163c34590141eecb14 (patch) | |
| tree | 64706ef579f548b804d5b674b33f6b239c638d0f /src/backends/neon/NeonLayerSupport.hpp | |
| parent | e40cc8359b02a7786908294300c45b672cf6b0e4 (diff) | |
| download | armnn-9145e38edf49fa4862008c163c34590141eecb14.tar.gz | |
IVGCVSW-7505 Create FusedLayer and NeonFusedWorkload for AddMulAdd Neon kernel
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ic778d35b001474b44fb1e433a6fe276e4ec9f565
Diffstat (limited to 'src/backends/neon/NeonLayerSupport.hpp')
| -rw-r--r-- | src/backends/neon/NeonLayerSupport.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/backends/neon/NeonLayerSupport.hpp b/src/backends/neon/NeonLayerSupport.hpp index 4bc96acd30..0295c2b3e2 100644 --- a/src/backends/neon/NeonLayerSupport.hpp +++ b/src/backends/neon/NeonLayerSupport.hpp @@ -151,6 +151,11 @@ public: const FullyConnectedDescriptor& descriptor, Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const; + bool IsFusedSupported(const std::vector<std::reference_wrapper<TensorInfo>>& inputs, + const std::vector<std::reference_wrapper<TensorInfo>>& outputs, + const FusedDescriptor& descriptor, + Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const; + bool IsGatherNdSupported(const TensorInfo& input0, const TensorInfo& input1, const TensorInfo& output, |