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 /include/armnn/Descriptors.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 'include/armnn/Descriptors.hpp')
-rw-r--r-- | include/armnn/Descriptors.hpp | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index f60e8f3bea..30eaefd83b 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -940,6 +940,27 @@ struct FillDescriptor : BaseDescriptor float m_Value; }; +/// A FusedDescriptor for the FusedLayer. +struct FusedDescriptor : BaseDescriptor +{ + FusedDescriptor(unsigned int numInputSlots = 4u, + unsigned int numOutputSlots = 2u, + FusedKernelType fusedType = FusedKernelType::AddMulAdd) + : m_NumInputSlots(numInputSlots), m_NumOutputSlots(numOutputSlots), m_FusedKernelType(fusedType) + {} + + bool operator ==(const FusedDescriptor& rhs) const + { + return m_NumInputSlots == rhs.m_NumInputSlots && + m_NumOutputSlots == rhs.m_NumOutputSlots && + m_FusedKernelType == rhs.m_FusedKernelType; + } + + unsigned int m_NumInputSlots; + unsigned int m_NumOutputSlots; + FusedKernelType m_FusedKernelType; +}; + /// A GatherDescriptor for the GatherLayer. struct GatherDescriptor : BaseDescriptor { |