diff options
Diffstat (limited to 'src/armnn/layers/ReshapeLayer.hpp')
| -rw-r--r-- | src/armnn/layers/ReshapeLayer.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/layers/ReshapeLayer.hpp b/src/armnn/layers/ReshapeLayer.hpp index 4fd5c3e11d..5e0e883822 100644 --- a/src/armnn/layers/ReshapeLayer.hpp +++ b/src/armnn/layers/ReshapeLayer.hpp @@ -6,6 +6,8 @@ #include "LayerWithParameters.hpp" +#include <armnn/utility/PolymorphicDowncast.hpp> + namespace armnn { @@ -39,7 +41,7 @@ public: bool IsEqual(const Layer& other) const { return (other.GetType() == LayerType::Reshape) && - m_Param.m_TargetShape == boost::polymorphic_downcast<const ReshapeLayer*>(&other)->m_Param.m_TargetShape; + m_Param.m_TargetShape == PolymorphicDowncast<const ReshapeLayer*>(&other)->m_Param.m_TargetShape; } void Accept(ILayerVisitor& visitor) const override; |