diff options
Diffstat (limited to 'src/armnn/Layer.hpp')
-rw-r--r-- | src/armnn/Layer.hpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/src/armnn/Layer.hpp b/src/armnn/Layer.hpp index 114d69c652..767cf97eb3 100644 --- a/src/armnn/Layer.hpp +++ b/src/armnn/Layer.hpp @@ -238,6 +238,7 @@ public: } ShapeInferenceMethod GetShapeInferenceMethod() const { return m_ShapeInferenceMethod; }; + bool GetAllowExpandedDims() const { return m_AllowExpandedDims; }; const std::vector<InputSlot>& GetInputSlots() const { return m_InputSlots; } const std::vector<OutputSlot>& GetOutputSlots() const { return m_OutputSlots; } @@ -343,6 +344,11 @@ public: m_ShapeInferenceMethod = shapeInferenceMethod; } + void SetAllowExpandedDims(bool allowExpandedDims) + { + m_AllowExpandedDims = allowExpandedDims; + } + template<typename T> std::shared_ptr<T> GetAdditionalInformation() const { @@ -428,6 +434,8 @@ private: mutable LayerPriority m_Priority = 0; mutable bool m_Visiting = false; + bool m_AllowExpandedDims = false; + LayerGuid m_Guid; std::list<std::string> m_RelatedLayerNames; |