1 // 2 // Copyright © 2017 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 #include "TestInputOutputLayerVisitor.hpp" 6 #include "Network.hpp" 7 8 namespace armnn 9 { 10 11 BOOST_AUTO_TEST_SUITE(TestInputOutputLayerVisitor) 12 BOOST_AUTO_TEST_CASE(CheckInputLayerVisitorBindingIdAndName)13BOOST_AUTO_TEST_CASE(CheckInputLayerVisitorBindingIdAndName) 14 { 15 const char* layerName = "InputLayer"; 16 TestInputLayerVisitor visitor(1, layerName); 17 Network net; 18 19 IConnectableLayer *const layer = net.AddInputLayer(1, layerName); 20 layer->Accept(visitor); 21 } 22 BOOST_AUTO_TEST_CASE(CheckInputLayerVisitorBindingIdAndNameNull)23BOOST_AUTO_TEST_CASE(CheckInputLayerVisitorBindingIdAndNameNull) 24 { 25 TestInputLayerVisitor visitor(1); 26 Network net; 27 28 IConnectableLayer *const layer = net.AddInputLayer(1); 29 layer->Accept(visitor); 30 } 31 BOOST_AUTO_TEST_CASE(CheckOutputLayerVisitorBindingIdAndName)32BOOST_AUTO_TEST_CASE(CheckOutputLayerVisitorBindingIdAndName) 33 { 34 const char* layerName = "OutputLayer"; 35 TestOutputLayerVisitor visitor(1, layerName); 36 Network net; 37 38 IConnectableLayer *const layer = net.AddOutputLayer(1, layerName); 39 layer->Accept(visitor); 40 } 41 BOOST_AUTO_TEST_CASE(CheckOutputLayerVisitorBindingIdAndNameNull)42BOOST_AUTO_TEST_CASE(CheckOutputLayerVisitorBindingIdAndNameNull) 43 { 44 TestOutputLayerVisitor visitor(1); 45 Network net; 46 47 IConnectableLayer *const layer = net.AddOutputLayer(1); 48 layer->Accept(visitor); 49 } 50 51 BOOST_AUTO_TEST_SUITE_END() 52 53 } //namespace armnn