diff options
Diffstat (limited to 'src/armnn/optimizations/SquashEqualSiblings.hpp')
| -rw-r--r-- | src/armnn/optimizations/SquashEqualSiblings.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/optimizations/SquashEqualSiblings.hpp b/src/armnn/optimizations/SquashEqualSiblings.hpp index bac27c06a7..d836a9c549 100644 --- a/src/armnn/optimizations/SquashEqualSiblings.hpp +++ b/src/armnn/optimizations/SquashEqualSiblings.hpp @@ -7,6 +7,7 @@ #include "Optimization.hpp" #include <armnn/utility/IgnoreUnused.hpp> +#include <armnn/utility/PolymorphicDowncast.hpp> namespace armnn { @@ -32,7 +33,7 @@ public: if (baseOutput.GetNumConnections() > 1) { - auto& comparableChild = *boost::polymorphic_downcast<Comparable*>(&child); + auto& comparableChild = *PolymorphicDowncast<Comparable*>(&child); Layer* lowestPriorityChild = &child; for (auto&& it : baseOutput.GetConnections()) |