diff options
Diffstat (limited to 'src/armnn/Graph.hpp')
-rw-r--r-- | src/armnn/Graph.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/Graph.hpp b/src/armnn/Graph.hpp index 00ab8deaa0..09461885ee 100644 --- a/src/armnn/Graph.hpp +++ b/src/armnn/Graph.hpp @@ -12,6 +12,7 @@ #include <armnn/NetworkFwd.hpp> #include <armnn/Exceptions.hpp> #include <armnn/utility/Assert.hpp> +#include <armnn/utility/PolymorphicDowncast.hpp> #include <list> #include <map> @@ -32,7 +33,7 @@ public: template <typename LayerType> static LayerType* PtrCast(Layer* const layer) { - return boost::polymorphic_downcast<LayerType*>(layer); + return PolymorphicDowncast<LayerType*>(layer); } template <typename Func> |