diff options
Diffstat (limited to 'src/armnnTfLiteParser/test/DetectionPostProcess.cpp')
-rw-r--r-- | src/armnnTfLiteParser/test/DetectionPostProcess.cpp | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/src/armnnTfLiteParser/test/DetectionPostProcess.cpp b/src/armnnTfLiteParser/test/DetectionPostProcess.cpp index f12b2b94d6..cb4173a620 100644 --- a/src/armnnTfLiteParser/test/DetectionPostProcess.cpp +++ b/src/armnnTfLiteParser/test/DetectionPostProcess.cpp @@ -4,16 +4,16 @@ // #include "../TfLiteParser.hpp" - -#include <boost/test/unit_test.hpp> -#include "test/GraphUtils.hpp" - #include "ParserFlatbuffersFixture.hpp" #include "ParserPrototxtFixture.hpp" #include "ParserHelper.hpp" +#include "test/GraphUtils.hpp" +#include <armnn/utility/PolymorphicDowncast.hpp> #include <QuantizeHelper.hpp> +#include <boost/test/unit_test.hpp> + BOOST_AUTO_TEST_SUITE(TensorflowLiteParser) struct DetectionPostProcessFixture : ParserFlatbuffersFixture @@ -241,7 +241,7 @@ BOOST_FIXTURE_TEST_CASE(DetectionPostProcessGraphStructureTest, ParseDetectionPo auto optimized = Optimize(*network, { armnn::Compute::CpuRef }, m_Runtime->GetDeviceSpec()); - auto optimizedNetwork = boost::polymorphic_downcast<armnn::OptimizedNetwork*>(optimized.get()); + auto optimizedNetwork = armnn::PolymorphicDowncast<armnn::OptimizedNetwork*>(optimized.get()); auto graph = optimizedNetwork->GetGraph(); // Check the number of layers in the graph |