From 3d2b4b2bff3be27f12a99e0e01284078870ee954 Mon Sep 17 00:00:00 2001 From: Francis Murtagh Date: Mon, 15 Feb 2021 18:23:17 +0000 Subject: IVGCVSW-4873 Implement Pimpl Idiom for INetwork and IOptimizedNetwork !android-nn-driver:5042 Signed-off-by: Kevin May Change-Id: Ia1ce8b839e81b46428ba0f78463e085e5906958d Signed-off-by: Francis Murtagh Signed-off-by: Finn Williams --- src/armnnTfParser/test/Assert.cpp | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) (limited to 'src/armnnTfParser/test') diff --git a/src/armnnTfParser/test/Assert.cpp b/src/armnnTfParser/test/Assert.cpp index b978f0264d..0665be7c7e 100644 --- a/src/armnnTfParser/test/Assert.cpp +++ b/src/armnnTfParser/test/Assert.cpp @@ -102,8 +102,7 @@ BOOST_FIXTURE_TEST_CASE(AssertSimpleGraphStructureTest, AssertSimpleFixture) { auto optimized = SetupOptimizedNetwork({ { "Placeholder", { 1, 1, 1, 4 } } }, { "Add" }); - auto optimizedNetwork = armnn::PolymorphicDowncast(optimized.get()); - auto graph = optimizedNetwork->GetGraph(); + armnn::Graph& graph = GetGraphForTesting(optimized.get()); BOOST_TEST((graph.GetNumInputs() == 1)); BOOST_TEST((graph.GetNumOutputs() == 1)); @@ -258,8 +257,7 @@ BOOST_FIXTURE_TEST_CASE(AssertGraphStructureTest, AssertFixture) { "Input1", { 1, 1, 2, 2 } } }, { "Output" }); - auto optimizedNetwork = armnn::PolymorphicDowncast(optimized.get()); - auto graph = optimizedNetwork->GetGraph(); + armnn::Graph& graph = GetGraphForTesting(optimized.get()); BOOST_TEST((graph.GetNumInputs() == 2)); BOOST_TEST((graph.GetNumOutputs() == 1)); -- cgit v1.2.1