diff options
Diffstat (limited to 'src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp')
-rw-r--r-- | src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp b/src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp index 222414c8c5..ca42cacb39 100644 --- a/src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp +++ b/src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp @@ -7,6 +7,8 @@ #include "NetworkUtils.hpp" #include "Optimization.hpp" +#include <armnn/utility/PolymorphicDowncast.hpp> + namespace armnn { namespace optimizations @@ -15,7 +17,7 @@ namespace optimizations template <typename LayerT> inline LayerT* ConvertWeight(Layer* l) { - LayerT* layer = boost::polymorphic_downcast<LayerT*>(l); + LayerT* layer = PolymorphicDowncast<LayerT*>(l); if ((layer->GetType() == LayerType::Convolution2d || layer->GetType() == LayerType::FullyConnected) && layer->m_Weight) { |