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