diff options
Diffstat (limited to 'src/armnnTfLiteParser/test')
-rw-r--r-- | src/armnnTfLiteParser/test/DetectionPostProcess.cpp | 10 | ||||
-rw-r--r-- | src/armnnTfLiteParser/test/Unsupported.cpp | 4 |
2 files changed, 7 insertions, 7 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 diff --git a/src/armnnTfLiteParser/test/Unsupported.cpp b/src/armnnTfLiteParser/test/Unsupported.cpp index 21392ace02..dd77bcacb1 100644 --- a/src/armnnTfLiteParser/test/Unsupported.cpp +++ b/src/armnnTfLiteParser/test/Unsupported.cpp @@ -8,10 +8,10 @@ #include <armnn/LayerVisitorBase.hpp> #include <armnn/utility/Assert.hpp> +#include <armnn/utility/PolymorphicDowncast.hpp> #include <layers/StandInLayer.hpp> -#include <boost/polymorphic_cast.hpp> #include <boost/test/unit_test.hpp> #include <sstream> @@ -47,7 +47,7 @@ public: BOOST_CHECK(descriptor.m_NumOutputs == numOutputs); BOOST_CHECK(layer->GetNumOutputSlots() == numOutputs); - const StandInLayer* standInLayer = boost::polymorphic_downcast<const StandInLayer*>(layer); + const StandInLayer* standInLayer = PolymorphicDowncast<const StandInLayer*>(layer); for (unsigned int i = 0u; i < numInputs; ++i) { const OutputSlot* connectedSlot = standInLayer->GetInputSlot(i).GetConnectedOutputSlot(); |