diff options
Diffstat (limited to 'src/armnn/optimizations/MovePermuteUp.hpp')
| -rw-r--r-- | src/armnn/optimizations/MovePermuteUp.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/optimizations/MovePermuteUp.hpp b/src/armnn/optimizations/MovePermuteUp.hpp index a7a477be84..15c6f61e97 100644 --- a/src/armnn/optimizations/MovePermuteUp.hpp +++ b/src/armnn/optimizations/MovePermuteUp.hpp @@ -6,6 +6,7 @@ #include "Optimization.hpp" +#include <armnn/utility/PolymorphicDowncast.hpp> #include <armnnUtils/Permute.hpp> namespace armnn @@ -29,7 +30,7 @@ public: if (CanMovePermuteToInputs(base)) { - auto permute = boost::polymorphic_downcast<PermuteLayer*>(&connection.GetOwningLayer()); + auto permute = PolymorphicDowncast<PermuteLayer*>(&connection.GetOwningLayer()); const PermutationVector& perm = permute->GetPermutation(); // Inserts an equivalent permute before every input of the base layer. |