src/armnnCaffeParser/test/TestSplit.cpp

//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include <boost/test/unit_test.hpp>
#include "armnnCaffeParser/ICaffeParser.hpp"
#include "ParserPrototxtFixture.hpp"

BOOST_AUTO_TEST_SUITE(CaffeParser)

struct SplitFixture : public armnnUtils::ParserPrototxtFixture<armnnCaffeParser::ICaffeParser>
{
    SplitFixture()
    {
        m_Prototext = R"(
name: "Split"
layer {
  name: "data"
  type: "Input"
  top: "data"
  input_param { shape: { dim: 1 dim: 1 dim: 1 dim: 1 } }
}
layer {
    name: "split"
    type: "Split"
    bottom: "data"
    top: "split_top0"
    top: "split_top1"
}
layer {
    bottom: "split_top0"
    bottom: "split_top1"
    top: "add"
    name: "add"
    type: "Eltwise"
}
        )";
        SetupSingleInputSingleOutput("data", "add");
    }
};

BOOST_FIXTURE_TEST_CASE(Split, SplitFixture)
{
    RunTest<1>({ 1.0f }, { 2.0f });
}

BOOST_AUTO_TEST_SUITE_END()