diff options
Diffstat (limited to 'src/armnn/optimizations/Optimization.hpp')
-rw-r--r-- | src/armnn/optimizations/Optimization.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/optimizations/Optimization.hpp b/src/armnn/optimizations/Optimization.hpp index efe3930db8..565f543bee 100644 --- a/src/armnn/optimizations/Optimization.hpp +++ b/src/armnn/optimizations/Optimization.hpp @@ -7,6 +7,8 @@ #include "Graph.hpp" #include "LayersFwd.hpp" +#include <armnn/utility/PolymorphicDowncast.hpp> + namespace armnn { @@ -37,7 +39,7 @@ public: { if (base.GetType() == LayerEnumOf<BaseType>()) { - Wrapped::Run(graph, *boost::polymorphic_downcast<BaseType*>(&base)); + Wrapped::Run(graph, *PolymorphicDowncast<BaseType*>(&base)); } } |