diff options
author | telsoa01 <telmo.soares@arm.com> | 2018-08-31 09:22:23 +0100 |
---|---|---|
committer | telsoa01 <telmo.soares@arm.com> | 2018-08-31 09:22:23 +0100 |
commit | c577f2c6a3b4ddb6ba87a882723c53a248afbeba (patch) | |
tree | bd7d4c148df27f8be6649d313efb24f536b7cf34 /include/armnn/Descriptors.hpp | |
parent | 4c7098bfeab1ffe1cdc77f6c15548d3e73274746 (diff) | |
download | armnn-c577f2c6a3b4ddb6ba87a882723c53a248afbeba.tar.gz |
Release 18.08
Diffstat (limited to 'include/armnn/Descriptors.hpp')
-rw-r--r-- | include/armnn/Descriptors.hpp | 32 |
1 files changed, 30 insertions, 2 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index 2595656c70..3cf152befe 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -95,8 +95,8 @@ private: uint32_t** m_ViewSizes; }; -// Convenience template to create a OriginsDescriptor to use when creating a Merger layer for performing concatenation -// of a number of input tensors +/// Convenience template to create an OriginsDescriptor to use when creating a Merger layer for performing concatenation +/// of a number of input tensors template <typename TensorShapeIt> OriginsDescriptor CreateMergerDescriptorForConcatenation(TensorShapeIt first, TensorShapeIt last, unsigned int concatenationDimension) @@ -301,7 +301,35 @@ struct ResizeBilinearDescriptor struct ReshapeDescriptor { + ReshapeDescriptor() + : m_TargetShape() + {} + + ReshapeDescriptor(const TensorShape& shape) + : m_TargetShape(shape) + {} + TensorShape m_TargetShape; }; +// temporary descriptor for Lstm +struct LstmDescriptor +{ + LstmDescriptor() + : m_ActivationFunc(1) // 0: None, 1: Relu, 3: Relu6, 4: Tanh, 6: Sigmoid + , m_ClippingThresCell(0.0) + , m_ClippingThresProj(0.0) + , m_CifgEnabled(true) + , m_PeepholeEnabled(false) + , m_ProjectionEnabled(false) + {} + + uint32_t m_ActivationFunc; + float m_ClippingThresCell; + float m_ClippingThresProj; + bool m_CifgEnabled; + bool m_PeepholeEnabled; + bool m_ProjectionEnabled; +}; + } |