1 // 2 // Copyright © 2017 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 #include <boost/test/unit_test.hpp> 6 #include "armnnCaffeParser/ICaffeParser.hpp" 7 #include "ParserPrototxtFixture.hpp" 8 9 BOOST_AUTO_TEST_SUITE(CaffeParser) 10 11 struct SplitFixture : public armnnUtils::ParserPrototxtFixture<armnnCaffeParser::ICaffeParser> 12 { SplitFixtureSplitFixture13 SplitFixture() 14 { 15 m_Prototext = R"( 16 name: "Split" 17 layer { 18 name: "data" 19 type: "Input" 20 top: "data" 21 input_param { shape: { dim: 1 dim: 1 dim: 1 dim: 1 } } 22 } 23 layer { 24 name: "split" 25 type: "Split" 26 bottom: "data" 27 top: "split_top0" 28 top: "split_top1" 29 } 30 layer { 31 bottom: "split_top0" 32 bottom: "split_top1" 33 top: "add" 34 name: "add" 35 type: "Eltwise" 36 } 37 )"; 38 SetupSingleInputSingleOutput("data", "add"); 39 } 40 }; 41 BOOST_FIXTURE_TEST_CASE(Split,SplitFixture)42BOOST_FIXTURE_TEST_CASE(Split, SplitFixture) 43 { 44 RunTest<1>({ 1.0f }, { 2.0f }); 45 } 46 47 BOOST_AUTO_TEST_SUITE_END() 48