TestMultiInputsOutputs.cpp

Go to the documentation of this file.

//
// 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 MultiInputsOutputsFixture : public armnnUtils::ParserPrototxtFixture<armnnCaffeParser::ICaffeParser>
{
    MultiInputsOutputsFixture()
    {
        m_Prototext = R"(
name: "MultiInputsOutputs"
layer {
  name: "input1"
  type: "Input"
  top: "input1"
  input_param { shape: { dim: 1 } }
}
layer {
  name: "input2"
  type: "Input"
  top: "input2"
  input_param { shape: { dim: 1 } }
}
layer {
    bottom: "input1"
    bottom: "input2"
    top: "add1"
    name: "add1"
    type: "Eltwise"
}
layer {
    bottom: "input2"
    bottom: "input1"
    top: "add2"
    name: "add2"
    type: "Eltwise"
}
        )";
        Setup({ }, { "add1", "add2" });
    }
};

BOOST_FIXTURE_TEST_CASE(MultiInputsOutputs, MultiInputsOutputsFixture)
{
    RunTest<1>({ { "input1",{ 12.0f } },{ "input2",{ 13.0f } } },
        { { "add1",{ 25.0f } },{ "add2",{ 25.0f } } });
}

BOOST_AUTO_TEST_SUITE_END()