diff options
Diffstat (limited to 'src/armnnCaffeParser/test')
-rw-r--r-- | src/armnnCaffeParser/test/TestAdd.cpp | 70 | ||||
-rw-r--r-- | src/armnnCaffeParser/test/TestConcat.cpp | 73 | ||||
-rw-r--r-- | src/armnnCaffeParser/test/TestConvolution.cpp | 133 | ||||
-rw-r--r-- | src/armnnCaffeParser/test/TestDropout.cpp | 53 | ||||
-rw-r--r-- | src/armnnCaffeParser/test/TestInPlace.cpp | 98 | ||||
-rw-r--r-- | src/armnnCaffeParser/test/TestInputs.cpp | 122 | ||||
-rw-r--r-- | src/armnnCaffeParser/test/TestMul.cpp | 73 | ||||
-rw-r--r-- | src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp | 54 | ||||
-rw-r--r-- | src/armnnCaffeParser/test/TestPooling2d.cpp | 54 | ||||
-rw-r--r-- | src/armnnCaffeParser/test/TestSplit.cpp | 47 |
10 files changed, 0 insertions, 777 deletions
diff --git a/src/armnnCaffeParser/test/TestAdd.cpp b/src/armnnCaffeParser/test/TestAdd.cpp deleted file mode 100644 index ab087cbb89..0000000000 --- a/src/armnnCaffeParser/test/TestAdd.cpp +++ /dev/null @@ -1,70 +0,0 @@ -// -// 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 AddFixture : public armnnUtils::ParserPrototxtFixture<armnnCaffeParser::ICaffeParser> -{ - AddFixture() - { - m_Prototext = "name: \"MinimalAdd\"\n" - "layer {\n" - " name: \"data\"\n" - " type: \"Input\"\n" - " top: \"data\"\n" - " input_param { shape: { dim: 1 dim: 1 dim: 4 dim: 4 } }\n" - "}\n" - "layer {\n" - " bottom: \"data\"\n" - " top: \"pool1\"\n" - " name: \"pool1\"\n" - " type: \"Pooling\"\n" - " pooling_param {\n" - " kernel_size: 2\n" - " stride: 2\n" - " pool: MAX\n" - " }\n" - "}\n" - "layer {\n" - " bottom: \"data\"\n" - " top: \"pool2\"\n" - " name: \"pool2\"\n" - " type: \"Pooling\"\n" - " pooling_param {\n" - " kernel_size: 2\n" - " stride: 2\n" - " pool: MAX\n" - " }\n" - "}\n" - "layer {\n" - " bottom: \"pool1\"\n" - " bottom: \"pool2\"\n" - " top: \"add\"\n" - " name: \"add\"\n" - " type: \"Eltwise\"\n" - "}\n"; - SetupSingleInputSingleOutput("data", "add"); - } -}; - -BOOST_FIXTURE_TEST_CASE(ParseAdd, AddFixture) -{ - RunTest<4>( - { - 0, 1, 0, 0, - 0, 0, 0, 0, - 0, 0, 1, 0, - 1, 0, 1, 1 - }, - { - 2, 0, - 2, 2 - }); -} - -BOOST_AUTO_TEST_SUITE_END() diff --git a/src/armnnCaffeParser/test/TestConcat.cpp b/src/armnnCaffeParser/test/TestConcat.cpp deleted file mode 100644 index 2d952865f4..0000000000 --- a/src/armnnCaffeParser/test/TestConcat.cpp +++ /dev/null @@ -1,73 +0,0 @@ -// -// 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 ConcatFixture : public armnnUtils::ParserPrototxtFixture<armnnCaffeParser::ICaffeParser> -{ - ConcatFixture() - { - m_Prototext = "name: \"Concat\"\n" - "layer {\n" - " name: \"data\"\n" - " type: \"Input\"\n" - " top: \"data\"\n" - " input_param { shape: { dim: 1 dim: 1 dim: 4 dim: 4 } }\n" - "}\n" - "layer {\n" - " bottom: \"data\"\n" - " top: \"pool1\"\n" - " name: \"pool1\"\n" - " type: \"Pooling\"\n" - " pooling_param {\n" - " kernel_size: 2\n" - " stride: 2\n" - " pool: MAX\n" - " }\n" - "}\n" - "layer {\n" - " bottom: \"data\"\n" - " top: \"pool2\"\n" - " name: \"pool2\"\n" - " type: \"Pooling\"\n" - " pooling_param {\n" - " kernel_size: 2\n" - " stride: 2\n" - " pool: MAX\n" - " }\n" - "}\n" - "layer {\n" - " bottom: \"pool1\"\n" - " bottom: \"pool2\"\n" - " top: \"concat\"\n" - " name: \"concat\"\n" - " type: \"Concat\"\n" - "}\n"; - SetupSingleInputSingleOutput("data", "concat"); - } -}; - -BOOST_FIXTURE_TEST_CASE(ParseConcat, ConcatFixture) -{ - RunTest<4>( - { - 0, 1, 0, 0, - 0, 0, 0, 0, - 0, 0, 1, 0, - 1, 0, 1, 1 - }, - { - 1, 0, - 1, 1, - - 1, 0, - 1, 1 - }); -} - -BOOST_AUTO_TEST_SUITE_END() diff --git a/src/armnnCaffeParser/test/TestConvolution.cpp b/src/armnnCaffeParser/test/TestConvolution.cpp deleted file mode 100644 index b881f1f58f..0000000000 --- a/src/armnnCaffeParser/test/TestConvolution.cpp +++ /dev/null @@ -1,133 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// -#include <boost/test/unit_test.hpp> -#include "armnnCaffeParser/ICaffeParser.hpp" -#include "ParserPrototxtFixture.hpp" -#include <sstream> -#include <initializer_list> - -namespace -{ - -template <typename T> -std::string TaggedSequence(const std::string & tag, const std::initializer_list<T> & data) -{ - bool first = true; - std::stringstream ss; - for (auto && d : data) - { - if (!first) - { - ss << " , "; - } - else - { - first = false; - } - ss << " " << tag << " : " << d << " "; - } - return ss.str(); -} - -template <typename T> -std::string TaggedSequence(const std::string & tag, T data, unsigned int n) -{ - std::stringstream ss; - for (unsigned int i=0; i<n; ++i) - { - if (i>0) - { - ss << " , "; - } - ss << " " << tag << " : " << data << " "; - } - return ss.str(); -} - -} // namespace <anonymous> - -BOOST_AUTO_TEST_SUITE(CaffeParser) - -struct ConvolutionFixture : public armnnUtils::ParserPrototxtFixture<armnnCaffeParser::ICaffeParser> -{ - ConvolutionFixture(const std::initializer_list<unsigned int> & inputDims, - const std::initializer_list<float> & filterData, - unsigned int kernelSize, - unsigned int numOutput=1, - unsigned int group=1) - { - m_Prototext = R"( - name: "ConvolutionTest" - layer { - name: "input1" - type: "Input" - top: "input1" - input_param { shape: { )" + TaggedSequence("dim", inputDims) + R"( } } - } - layer { - name: "conv1" - type: "Convolution" - bottom: "input1" - top: "conv1" - blobs: { )" + TaggedSequence("data", filterData) + R"( } - blobs: { )" + TaggedSequence("data", 0, numOutput) + R"( } - convolution_param { - num_output: )" + std::to_string(numOutput) + R"( - pad: 0 - kernel_size: )" + std::to_string(kernelSize) + R"( - stride: 1 - group: )" + std::to_string(group) + R"( - } - } - )"; - SetupSingleInputSingleOutput("input1", "conv1"); - } -}; - -struct SimpleConvolutionFixture : public ConvolutionFixture -{ - SimpleConvolutionFixture() - : ConvolutionFixture( {1, 1, 2, 2}, {1.0f, 1.0f, 1.0f, 1.0f}, 2) - { - } -}; - -BOOST_FIXTURE_TEST_CASE(SimpleConvolution, SimpleConvolutionFixture) -{ - RunTest<4>({ 1, 3, 5, 7 }, { 16 }); -} - -struct GroupConvolutionFixture : public ConvolutionFixture -{ - GroupConvolutionFixture() - : ConvolutionFixture( - {1, 2, 2, 2}, - { - 1.0f, 1.0f, 1.0f, 1.0f, // filter for channel #0 - 2.0f, 2.0f, 2.0f, 2.0f // filter for channel #1 - }, - 2, // kernel size is 2x2 - 2, // number of output channels is 2 - 2) // number of groups (separate filters) - { - } -}; - -BOOST_FIXTURE_TEST_CASE(GroupConvolution, GroupConvolutionFixture) -{ - RunTest<4>( - { - 1, 2, 3, 4, // input channel #0 - 5, 6, 7, 8, // input channel #1 - }, - { - 10, // convolution result for channel #0 applying filter #0 - 52 // same for channel #1 and filter #1 - } - ); -} - - -BOOST_AUTO_TEST_SUITE_END()
\ No newline at end of file diff --git a/src/armnnCaffeParser/test/TestDropout.cpp b/src/armnnCaffeParser/test/TestDropout.cpp deleted file mode 100644 index 503766248b..0000000000 --- a/src/armnnCaffeParser/test/TestDropout.cpp +++ /dev/null @@ -1,53 +0,0 @@ -// -// 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 DropoutFixture : public armnnUtils::ParserPrototxtFixture<armnnCaffeParser::ICaffeParser> -{ - DropoutFixture() - { - m_Prototext = "name: \"DropoutFixture\"\n" - "layer {\n" - " name: \"data\"\n" - " type: \"Input\"\n" - " top: \"data\"\n" - " input_param { shape: { dim: 1 dim: 1 dim: 2 dim: 2 } }\n" - "}\n" - "layer {\n" - " bottom: \"data\"\n" - " top: \"drop1\"\n" - " name: \"drop1\"\n" - " type: \"Dropout\"\n" - "}\n" - "layer {\n" - " bottom: \"drop1\"\n" - " bottom: \"drop1\"\n" - " top: \"add\"\n" - " name: \"add\"\n" - " type: \"Eltwise\"\n" - "}\n"; - SetupSingleInputSingleOutput("data", "add"); - } -}; - -BOOST_FIXTURE_TEST_CASE(ParseDropout, DropoutFixture) -{ - RunTest<4>( - { - 1, 2, - 3, 4, - }, - { - 2, 4, - 6, 8 - }); -} - -BOOST_AUTO_TEST_SUITE_END() diff --git a/src/armnnCaffeParser/test/TestInPlace.cpp b/src/armnnCaffeParser/test/TestInPlace.cpp deleted file mode 100644 index 2495e2182e..0000000000 --- a/src/armnnCaffeParser/test/TestInPlace.cpp +++ /dev/null @@ -1,98 +0,0 @@ -// -// 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) - -// The pooling layer should take its input from the relu, not the add directly. -struct InPlaceFixture : public armnnUtils::ParserPrototxtFixture<armnnCaffeParser::ICaffeParser> -{ - InPlaceFixture() - { - m_Prototext = R"( -name: "InPlace" -layer { - name: "data" - type: "Input" - top: "data" - input_param { shape: { dim: 1 dim: 1 dim: 1 dim: 1 } } -} -layer { - bottom: "data" - bottom: "data" - top: "add" - name: "add" - type: "Eltwise" -} -layer { - name: "relu" - type: "ReLU" - bottom: "add" - top: "relu" - phase: TEST -} -layer { - name: "pool" - type: "Pooling" - bottom: "relu" - top: "pool" - phase: TEST - pooling_param { - pool: MAX - kernel_size: 1 - stride: 1 - } -} - )"; - SetupSingleInputSingleOutput("data", "pool"); - } -}; - -BOOST_FIXTURE_TEST_CASE(ParseInPlace, InPlaceFixture) -{ - RunTest<1>({ -1.0f }, { 0.0f }); -} - -// The requested output of the network is a layer which has an activation attached. -// The output of the network should therefore actually be the activation layer. -struct InPlaceOutputFixture : public armnnUtils::ParserPrototxtFixture<armnnCaffeParser::ICaffeParser> -{ - InPlaceOutputFixture() - { - m_Prototext = R"( -name: "InPlace" -layer { - name: "data" - type: "Input" - top: "data" - input_param { shape: { dim: 1 dim: 1 dim: 1 dim: 1 } } -} -layer { - bottom: "data" - bottom: "data" - top: "add" - name: "add" - type: "Eltwise" -} -layer { - name: "relu" - type: "ReLU" - bottom: "add" - top: "add" - phase: TEST -} - )"; - SetupSingleInputSingleOutput("data", "add"); - } -}; - -BOOST_FIXTURE_TEST_CASE(InPlaceOutput, InPlaceOutputFixture) -{ - RunTest<1>({ -1.0f }, { 0.0f }); -} - -BOOST_AUTO_TEST_SUITE_END() diff --git a/src/armnnCaffeParser/test/TestInputs.cpp b/src/armnnCaffeParser/test/TestInputs.cpp deleted file mode 100644 index 96d8e2b8af..0000000000 --- a/src/armnnCaffeParser/test/TestInputs.cpp +++ /dev/null @@ -1,122 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// -#include <boost/test/unit_test.hpp> -#include "armnnCaffeParser/ICaffeParser.hpp" -#include "armnn/IRuntime.hpp" -#include "armnn/INetwork.hpp" -#include "armnn/Exceptions.hpp" - -#include "test/TensorHelpers.hpp" - -#include <string> - -#include "ParserPrototxtFixture.hpp" - -BOOST_AUTO_TEST_SUITE(CaffeParser) - - -BOOST_AUTO_TEST_CASE(InputShapes) -{ - std::string explicitInput = "name: \"Minimal\"\n" - "layer {\n" - " name: \"data\"\n" - " type: \"Input\"\n" - " top: \"data\"\n" - " input_param { shape: { dim: 1 dim: 2 dim: 3 dim: 4 } }\n" - "}"; - std::string implicitInput = "name: \"Minimal\"\n" - "input: \"data\" \n" - "input_dim: 1 \n" - "input_dim: 2 \n" - "input_dim: 3 \n" - "input_dim: 4 \n"; - std::string implicitInputNoShape = "name: \"Minimal\"\n" - "input: \"data\" \n"; - - armnn::IRuntime::CreationOptions options; - armnn::IRuntimePtr runtime(armnn::IRuntime::Create(options)); - armnnCaffeParser::ICaffeParserPtr parser(armnnCaffeParser::ICaffeParser::Create()); - armnn::INetworkPtr network(nullptr, nullptr); - armnn::NetworkId netId; - - // Check everything works normally - std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef}; - { - network = parser->CreateNetworkFromString(explicitInput.c_str(), {}, { "data" }); - BOOST_TEST(network.get()); - runtime->LoadNetwork(netId, Optimize(*network, backends, runtime->GetDeviceSpec())); - - armnnCaffeParser::BindingPointInfo inputBindingInfo = parser->GetNetworkInputBindingInfo("data"); - armnn::TensorInfo inputTensorInfo = inputBindingInfo.second; - BOOST_TEST((inputTensorInfo == runtime->GetInputTensorInfo(netId, inputBindingInfo.first))); - - BOOST_TEST(inputTensorInfo.GetShape()[0] == 1); - BOOST_TEST(inputTensorInfo.GetShape()[1] == 2); - BOOST_TEST(inputTensorInfo.GetShape()[2] == 3); - BOOST_TEST(inputTensorInfo.GetShape()[3] == 4); - } - - // Checks everything works with implicit input. - { - network = parser->CreateNetworkFromString(implicitInput.c_str(), {}, { "data" }); - BOOST_TEST(network.get()); - runtime->LoadNetwork(netId, Optimize(*network, backends, runtime->GetDeviceSpec())); - - armnnCaffeParser::BindingPointInfo inputBindingInfo = parser->GetNetworkInputBindingInfo("data"); - armnn::TensorInfo inputTensorInfo = inputBindingInfo.second; - BOOST_TEST((inputTensorInfo == runtime->GetInputTensorInfo(netId, inputBindingInfo.first))); - - BOOST_TEST(inputTensorInfo.GetShape()[0] == 1); - BOOST_TEST(inputTensorInfo.GetShape()[1] == 2); - BOOST_TEST(inputTensorInfo.GetShape()[2] == 3); - BOOST_TEST(inputTensorInfo.GetShape()[3] == 4); - } - - // Checks everything works with implicit and passing shape. - { - network = parser->CreateNetworkFromString(implicitInput.c_str(), { {"data", { 2, 2, 3, 4 } } }, { "data" }); - BOOST_TEST(network.get()); - runtime->LoadNetwork(netId, Optimize(*network, backends, runtime->GetDeviceSpec())); - - armnnCaffeParser::BindingPointInfo inputBindingInfo = parser->GetNetworkInputBindingInfo("data"); - armnn::TensorInfo inputTensorInfo = inputBindingInfo.second; - BOOST_TEST((inputTensorInfo == runtime->GetInputTensorInfo(netId, inputBindingInfo.first))); - - BOOST_TEST(inputTensorInfo.GetShape()[0] == 2); - BOOST_TEST(inputTensorInfo.GetShape()[1] == 2); - BOOST_TEST(inputTensorInfo.GetShape()[2] == 3); - BOOST_TEST(inputTensorInfo.GetShape()[3] == 4); - } - - // Checks everything works with implicit (no shape) and passing shape. - { - network = parser->CreateNetworkFromString(implicitInputNoShape.c_str(), {{"data", {2, 2, 3, 4} }}, { "data" }); - BOOST_TEST(network.get()); - runtime->LoadNetwork(netId, Optimize(*network, backends, runtime->GetDeviceSpec())); - - armnnCaffeParser::BindingPointInfo inputBindingInfo = parser->GetNetworkInputBindingInfo("data"); - armnn::TensorInfo inputTensorInfo = inputBindingInfo.second; - BOOST_TEST((inputTensorInfo == runtime->GetInputTensorInfo(netId, inputBindingInfo.first))); - - BOOST_TEST(inputTensorInfo.GetShape()[0] == 2); - BOOST_TEST(inputTensorInfo.GetShape()[1] == 2); - BOOST_TEST(inputTensorInfo.GetShape()[2] == 3); - BOOST_TEST(inputTensorInfo.GetShape()[3] == 4); - } - - // Checks exception on incompatible shapes. - { - BOOST_CHECK_THROW(parser->CreateNetworkFromString(implicitInput.c_str(), {{"data",{ 2, 2, 3, 2 }}}, {"data"}), - armnn::ParseException); - } - - // Checks exception when no shape available. - { - BOOST_CHECK_THROW(parser->CreateNetworkFromString(implicitInputNoShape.c_str(), {}, { "data" }), - armnn::ParseException); - } -} - -BOOST_AUTO_TEST_SUITE_END() diff --git a/src/armnnCaffeParser/test/TestMul.cpp b/src/armnnCaffeParser/test/TestMul.cpp deleted file mode 100644 index 3b49015bce..0000000000 --- a/src/armnnCaffeParser/test/TestMul.cpp +++ /dev/null @@ -1,73 +0,0 @@ -// -// 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 MulFixture : public armnnUtils::ParserPrototxtFixture<armnnCaffeParser::ICaffeParser> -{ - MulFixture() - { - m_Prototext = "name: \"MinimalMul\"\n" - "layer {\n" - " name: \"data\"\n" - " type: \"Input\"\n" - " top: \"data\"\n" - " input_param { shape: { dim: 1 dim: 1 dim: 4 dim: 4 } }\n" - "}\n" - "layer {\n" - " bottom: \"data\"\n" - " top: \"pool1\"\n" - " name: \"pool1\"\n" - " type: \"Pooling\"\n" - " pooling_param {\n" - " kernel_size: 2\n" - " stride: 2\n" - " pool: MAX\n" - " }\n" - "}\n" - "layer {\n" - " bottom: \"data\"\n" - " top: \"pool2\"\n" - " name: \"pool2\"\n" - " type: \"Pooling\"\n" - " pooling_param {\n" - " kernel_size: 2\n" - " stride: 2\n" - " pool: MAX\n" - " }\n" - "}\n" - "layer {\n" - " bottom: \"pool1\"\n" - " bottom: \"pool2\"\n" - " top: \"mul\"\n" - " name: \"mul\"\n" - " type: \"Eltwise\"\n" - " eltwise_param {\n" - " operation: 0\n" - " }\n" - "}\n"; - SetupSingleInputSingleOutput("data", "mul"); - } -}; - -BOOST_FIXTURE_TEST_CASE(ParseMul, MulFixture) -{ - RunTest<4>( - { - 0, 1, 0, 0, - 0, 0, 0, 0, - 0, 0, 1, 0, - 1, 0, 1, 1 - }, - { - 1, 0, - 1, 1 - }); -} - -BOOST_AUTO_TEST_SUITE_END() diff --git a/src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp b/src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp deleted file mode 100644 index 82b75f400e..0000000000 --- a/src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp +++ /dev/null @@ -1,54 +0,0 @@ -// -// 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() diff --git a/src/armnnCaffeParser/test/TestPooling2d.cpp b/src/armnnCaffeParser/test/TestPooling2d.cpp deleted file mode 100644 index 55517a0695..0000000000 --- a/src/armnnCaffeParser/test/TestPooling2d.cpp +++ /dev/null @@ -1,54 +0,0 @@ -// -// 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 GlobalPoolingFixture : public armnnUtils::ParserPrototxtFixture<armnnCaffeParser::ICaffeParser> -{ - GlobalPoolingFixture() - { - m_Prototext = "name: \"GlobalPooling\"\n" - "layer {\n" - " name: \"data\"\n" - " type: \"Input\"\n" - " top: \"data\"\n" - " input_param { shape: { dim: 1 dim: 3 dim: 2 dim: 2 } }\n" - "}\n" - "layer {\n" - " bottom: \"data\"\n" - " top: \"pool1\"\n" - " name: \"pool1\"\n" - " type: \"Pooling\"\n" - " pooling_param {\n" - " pool: AVE\n" - " global_pooling: true\n" - " }\n" - "}\n"; - SetupSingleInputSingleOutput("data", "pool1"); - } -}; - -BOOST_FIXTURE_TEST_CASE(GlobalPooling, GlobalPoolingFixture) -{ - RunTest<4>( - { - 1,3, - 5,7, - - 2,2, - 2,2, - - 4,4, - 6,6 - }, - { - 4, 2, 5 - }); -} - -BOOST_AUTO_TEST_SUITE_END() diff --git a/src/armnnCaffeParser/test/TestSplit.cpp b/src/armnnCaffeParser/test/TestSplit.cpp deleted file mode 100644 index 048da424cf..0000000000 --- a/src/armnnCaffeParser/test/TestSplit.cpp +++ /dev/null @@ -1,47 +0,0 @@ -// -// 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() |