diff options
Diffstat (limited to 'src/armnn/test/TestInputOutputLayerVisitor.cpp')
-rw-r--r-- | src/armnn/test/TestInputOutputLayerVisitor.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/armnn/test/TestInputOutputLayerVisitor.cpp b/src/armnn/test/TestInputOutputLayerVisitor.cpp index 35ffc55e55..6563517da1 100644 --- a/src/armnn/test/TestInputOutputLayerVisitor.cpp +++ b/src/armnn/test/TestInputOutputLayerVisitor.cpp @@ -14,7 +14,7 @@ BOOST_AUTO_TEST_CASE(CheckInputLayerVisitorBindingIdAndName) { const char* layerName = "InputLayer"; TestInputLayerVisitor visitor(1, layerName); - Network net; + NetworkImpl net; IConnectableLayer *const layer = net.AddInputLayer(1, layerName); layer->Accept(visitor); @@ -23,7 +23,7 @@ BOOST_AUTO_TEST_CASE(CheckInputLayerVisitorBindingIdAndName) BOOST_AUTO_TEST_CASE(CheckInputLayerVisitorBindingIdAndNameNull) { TestInputLayerVisitor visitor(1); - Network net; + NetworkImpl net; IConnectableLayer *const layer = net.AddInputLayer(1); layer->Accept(visitor); @@ -33,7 +33,7 @@ BOOST_AUTO_TEST_CASE(CheckOutputLayerVisitorBindingIdAndName) { const char* layerName = "OutputLayer"; TestOutputLayerVisitor visitor(1, layerName); - Network net; + NetworkImpl net; IConnectableLayer *const layer = net.AddOutputLayer(1, layerName); layer->Accept(visitor); @@ -42,7 +42,7 @@ BOOST_AUTO_TEST_CASE(CheckOutputLayerVisitorBindingIdAndName) BOOST_AUTO_TEST_CASE(CheckOutputLayerVisitorBindingIdAndNameNull) { TestOutputLayerVisitor visitor(1); - Network net; + NetworkImpl net; IConnectableLayer *const layer = net.AddOutputLayer(1); layer->Accept(visitor); |