aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/Descriptors.hpp
diff options
context:
space:
mode:
authorTeresa Charlin <teresa.charlinreyes@arm.com>2023-08-17 18:44:58 +0100
committerTeresa Charlin <teresa.charlinreyes@arm.com>2023-08-28 12:37:25 +0100
commit9145e38edf49fa4862008c163c34590141eecb14 (patch)
tree64706ef579f548b804d5b674b33f6b239c638d0f /include/armnn/Descriptors.hpp
parente40cc8359b02a7786908294300c45b672cf6b0e4 (diff)
downloadarmnn-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.hpp21
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
{