diff options
Diffstat (limited to 'src/armnn/layers/ReshapeLayer.hpp')
| -rw-r--r-- | src/armnn/layers/ReshapeLayer.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnn/layers/ReshapeLayer.hpp b/src/armnn/layers/ReshapeLayer.hpp index 13e73be4e4..c4743d17df 100644 --- a/src/armnn/layers/ReshapeLayer.hpp +++ b/src/armnn/layers/ReshapeLayer.hpp @@ -43,6 +43,8 @@ public: m_Param.m_TargetShape == boost::polymorphic_downcast<const ReshapeLayer*>(&other)->m_Param.m_TargetShape; } + void Accept(ILayerVisitor& visitor) const override; + protected: /// Constructor to create a ReshapeLayer. /// @param [in] param ReshapeDescriptor to configure the reshape operation. |