diff options
Diffstat (limited to 'src/armnn/layers/PermuteLayer.hpp')
-rw-r--r-- | src/armnn/layers/PermuteLayer.hpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/armnn/layers/PermuteLayer.hpp b/src/armnn/layers/PermuteLayer.hpp index 4984cf26ee..f2057d48e9 100644 --- a/src/armnn/layers/PermuteLayer.hpp +++ b/src/armnn/layers/PermuteLayer.hpp @@ -6,6 +6,8 @@ #include "LayerWithParameters.hpp" +#include <armnn/utility/PolymorphicDowncast.hpp> + namespace armnn { @@ -45,7 +47,7 @@ public: bool IsInverse(const Layer& other) const { return (other.GetType() == LayerType::Permute) && - GetPermutation().IsInverse(boost::polymorphic_downcast<const PermuteLayer*>(&other)->GetPermutation()); + GetPermutation().IsInverse(PolymorphicDowncast<const PermuteLayer*>(&other)->GetPermutation()); } /// Indicates if the other layer received is equal to this one. @@ -54,7 +56,7 @@ public: bool IsEqual(const Layer& other) const { return (other.GetType() == LayerType::Permute) && - GetPermutation().IsEqual(boost::polymorphic_downcast<const PermuteLayer*>(&other)->GetPermutation()); + GetPermutation().IsEqual(PolymorphicDowncast<const PermuteLayer*>(&other)->GetPermutation()); } void Accept(ILayerVisitor& visitor) const override; |