Convolution2d.cpp File Reference

#include <boost/test/unit_test.hpp>
#include "armnnTfParser/ITfParser.hpp"
#include "ParserPrototxtFixture.hpp"
#include <array>
#include <string>
#include <iostream>

Go to the source code of this file.

Functions
	BOOST_FIXTURE_TEST_CASE (ParseConv2dNhwcSame, Convolution2dNhwcSameFixture)
	BOOST_FIXTURE_TEST_CASE (ParseConv2dNchwSame, Convolution2dNchwSameFixture)
	BOOST_FIXTURE_TEST_CASE (ParseConv2dNhwcValid, Convolution2dNhwcValidFixture)
	BOOST_FIXTURE_TEST_CASE (ParseConv2dNchwValid, Convolution2dNchwValidFixture)
	BOOST_FIXTURE_TEST_CASE (ParseConv2dStride2NhwcSame, Convolution2dStride2NhwcSameFixture)
	BOOST_FIXTURE_TEST_CASE (ParseConv2dStride2NchwSame, Convolution2dStride2NchwSameFixture)
	BOOST_FIXTURE_TEST_CASE (ParseConv2dStride2NhwcValid, Convolution2dStride2NhwcValidFixture)
	BOOST_FIXTURE_TEST_CASE (ParseConv2dStride2NchwValid, Convolution2dStride2NchwValidFixture)
	BOOST_FIXTURE_TEST_CASE (ParseConv2dDilation1Nhwc, Convolution2dDilation1NhwcFixture)
	BOOST_FIXTURE_TEST_CASE (ParseConv2dDilation1Nchw, Convolution2dDilation1NchwFixture)
	BOOST_AUTO_TEST_CASE (ParseConv2dDilation2)

Function Documentation

BOOST_AUTO_TEST_CASE()

BOOST_AUTO_TEST_CASE

(

ParseConv2dDilation2

)

Definition at line 266 of file Convolution2d.cpp.

References BOOST_AUTO_TEST_SUITE_END(), and ITfParser::Create().

{
    const char* prototext = ""
        "node {\n"
        "  name: \"graphInput\"\n"
        "  op: \"Placeholder\"\n"
        "  attr {\n"
        "    key: \"dtype\"\n"
        "    value {\n"
        "      type: DT_FLOAT\n"
        "    }\n"
        "  }\n"
        "  attr {\n"
        "    key: \"shape\"\n"
        "    value {\n"
        "      shape {\n"
        "      }\n"
        "    }\n"
        "  }\n"
        "}\n"
        "node {\n"
        "  name: \"Const_1\"\n"
        "  op: \"Const\"\n"
        "  attr {\n"
        "    key: \"dtype\"\n"
        "    value {\n"
        "      type: DT_FLOAT\n"
        "    }\n"
        "  }\n"
        "  attr {\n"
        "    key: \"value\"\n"
        "    value {\n"
        "      tensor {\n"
        "        dtype: DT_FLOAT\n"
        "        tensor_shape {\n"
        "          dim {\n"
        "            size: 1\n"
        "          }\n"
        "          dim {\n"
        "            size: 3\n"
        "          }\n"
        "          dim {\n"
        "            size: 1\n"
        "          }\n"
        "          dim {\n"
        "            size: 1\n"
        "          }\n"
        "        }\n"
        "        tensor_content: \"\\000\\000\\000?\\000\\000\\200?\\000\\000\\000?\"\n"
        "      }\n"
        "    }\n"
        "  }\n"
        "}\n"
        "node {\n"
        "  name: \"potato\"\n"
        "  op: \"Conv2D\"\n"
        "  input: \"graphInput\"\n"
        "  input: \"Const_1\"\n"
        "  attr {\n"
        "    key: \"T\"\n"
        "    value {\n"
        "      type: DT_FLOAT\n"
        "    }\n"
        "  }\n"
        "  attr {\n"
        "    key: \"data_format\"\n"
        "    value {\n"
        "      s: \"NHWC\"\n"
        "    }\n"
        "  }\n"
        "  attr {\n"
        "    key: \"padding\"\n"
        "    value {\n"
        "      s: \"SAME\"\n"
        "    }\n"
        "  }\n"
        "  attr {\n"
        "    key: \"strides\"\n"
        "    value {\n"
        "      list {\n"
        "        i: 1\n"
        "        i: 1\n"
        "        i: 1\n"
        "        i: 1\n"
        "      }\n"
        "    }\n"
        "  }\n"
        "  attr {\n"
        "    key: \"dilations\"\n"
        "    value {\n"
        "      list {\n"
        "        i: 1\n"
        "        i: 2\n"
        "        i: 2\n"
        "        i: 1\n"
        "      }\n"
        "    }\n"
        "  }\n"
        "  attr {\n"
        "    key: \"use_cudnn_on_gpu\"\n"
        "    value {\n"
        "      b: false\n"
        "    }\n"
        "  }\n"
        "}\n";

    std::map<std::string, armnn::TensorShape> inputShapes;
    armnn::TensorShape tensorShape = { 1, 3, 3, 1 };
    inputShapes["graphInput"] = tensorShape;
    armnnTfParser::ITfParserPtr parser = armnnTfParser::ITfParser::Create();
    BOOST_CHECK_THROW(parser->CreateNetworkFromString(prototext, inputShapes, { "potato" }), armnn::ParseException);
}

BOOST_FIXTURE_TEST_CASE() [1/10]

BOOST_FIXTURE_TEST_CASE	(	ParseConv2dNhwcSame	,
		Convolution2dNhwcSameFixture
	)

Definition at line 175 of file Convolution2d.cpp.

{
    RunTest<4>({1, 2, 3, 4, 5, 6}, {2, 4, 4, 6.5f, 10 , 8.5f});
}

BOOST_FIXTURE_TEST_CASE() [2/10]

BOOST_FIXTURE_TEST_CASE	(	ParseConv2dNchwSame	,
		Convolution2dNchwSameFixture
	)

Definition at line 184 of file Convolution2d.cpp.

{
    RunTest<4>({1, 2, 3, 4, 5, 6}, {2, 4, 4, 6.5f, 10 , 8.5f});
}

BOOST_FIXTURE_TEST_CASE() [3/10]

BOOST_FIXTURE_TEST_CASE	(	ParseConv2dNhwcValid	,
		Convolution2dNhwcValidFixture
	)

Definition at line 194 of file Convolution2d.cpp.

{
    RunTest<4>({1, 2, 3, 4, 5, 6}, {4, 10});
}

BOOST_FIXTURE_TEST_CASE() [4/10]

BOOST_FIXTURE_TEST_CASE	(	ParseConv2dNchwValid	,
		Convolution2dNchwValidFixture
	)

Definition at line 203 of file Convolution2d.cpp.

{
    RunTest<4>({1, 2, 3, 4, 5, 6}, {4, 10});
}

BOOST_FIXTURE_TEST_CASE() [5/10]

BOOST_FIXTURE_TEST_CASE	(	ParseConv2dStride2NhwcSame	,
		Convolution2dStride2NhwcSameFixture
	)

Definition at line 213 of file Convolution2d.cpp.

{
    RunTest<4>({1, 2, 3, 4, 5, 6, 7, 8, 9}, {2, 4, 6.5, 8.5, 11, 13});
}

BOOST_FIXTURE_TEST_CASE() [6/10]

BOOST_FIXTURE_TEST_CASE	(	ParseConv2dStride2NchwSame	,
		Convolution2dStride2NchwSameFixture
	)

Definition at line 222 of file Convolution2d.cpp.

{
    RunTest<4>({1, 2, 3, 4, 5, 6, 7, 8, 9}, {2, 4, 6.5, 8.5, 11, 13});
}

BOOST_FIXTURE_TEST_CASE() [7/10]

BOOST_FIXTURE_TEST_CASE	(	ParseConv2dStride2NhwcValid	,
		Convolution2dStride2NhwcValidFixture
	)

Definition at line 232 of file Convolution2d.cpp.

{
    RunTest<4>({1, 2, 3, 4, 5, 6, 7, 8, 9}, {4, 10, 16});
}

BOOST_FIXTURE_TEST_CASE() [8/10]

BOOST_FIXTURE_TEST_CASE	(	ParseConv2dStride2NchwValid	,
		Convolution2dStride2NchwValidFixture
	)

Definition at line 241 of file Convolution2d.cpp.

{
    RunTest<4>({1, 2, 3, 4, 5, 6, 7, 8, 9}, {4, 10, 16});
}

BOOST_FIXTURE_TEST_CASE() [9/10]

BOOST_FIXTURE_TEST_CASE	(	ParseConv2dDilation1Nhwc	,
		Convolution2dDilation1NhwcFixture
	)

Definition at line 251 of file Convolution2d.cpp.

{
    RunTest<4>({1, 2, 3, 4, 5, 6}, {2, 4, 4, 6.5f, 10 , 8.5f});
}

BOOST_FIXTURE_TEST_CASE() [10/10]

BOOST_FIXTURE_TEST_CASE	(	ParseConv2dDilation1Nchw	,
		Convolution2dDilation1NchwFixture
	)

Definition at line 260 of file Convolution2d.cpp.

{
    RunTest<4>({1, 2, 3, 4, 5, 6}, {2, 4, 4, 6.5f, 10 , 8.5f});
}