blob: 2217a32edfd37ef12ee33f539b0d00d5326a97ac (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include "TestNameOnlyLayerVisitor.hpp"
#include <Network.hpp>
#include <boost/test/unit_test.hpp>
namespace
{
#define ADD_LAYER_METHOD_1_PARAM(name) net.Add##name##Layer("name##Layer")
#define ADD_LAYER_METHOD_2_PARAM(name) net.Add##name##Layer(armnn::name##Descriptor(), "name##Layer")
#define TEST_CASE_CHECK_LAYER_VISITOR_NAME(name, numParams) \
BOOST_AUTO_TEST_CASE(Check##name##LayerVisitorName) \
{ \
Test##name##LayerVisitor visitor("name##Layer"); \
armnn::Network net; \
armnn::IConnectableLayer *const layer = ADD_LAYER_METHOD_##numParams##_PARAM(name); \
layer->Accept(visitor); \
}
#define ADD_LAYER_METHOD_NULLPTR_1_PARAM(name) net.Add##name##Layer()
#define ADD_LAYER_METHOD_NULLPTR_2_PARAM(name) net.Add##name##Layer(armnn::name##Descriptor())
#define TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(name, numParams) \
BOOST_AUTO_TEST_CASE(Check##name##LayerVisitorNameNullptr) \
{ \
Test##name##LayerVisitor visitor; \
armnn::Network net; \
armnn::IConnectableLayer *const layer = ADD_LAYER_METHOD_NULLPTR_##numParams##_PARAM(name); \
layer->Accept(visitor); \
}
#define TEST_SUITE_NAME_ONLY_LAYER_VISITOR_1_PARAM(name) \
TEST_CASE_CHECK_LAYER_VISITOR_NAME(name, 1) \
TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(name, 1)
#define TEST_SUITE_NAME_ONLY_LAYER_VISITOR_2_PARAM(name) \
TEST_CASE_CHECK_LAYER_VISITOR_NAME(name, 2) \
TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(name, 2)
} // anonymous namespace
BOOST_AUTO_TEST_SUITE(TestNameOnlyLayerVisitor)
TEST_SUITE_NAME_ONLY_LAYER_VISITOR_1_PARAM(Addition)
TEST_SUITE_NAME_ONLY_LAYER_VISITOR_2_PARAM(DepthToSpace)
TEST_SUITE_NAME_ONLY_LAYER_VISITOR_1_PARAM(Division)
TEST_SUITE_NAME_ONLY_LAYER_VISITOR_1_PARAM(Equal)
TEST_SUITE_NAME_ONLY_LAYER_VISITOR_1_PARAM(Floor)
TEST_SUITE_NAME_ONLY_LAYER_VISITOR_1_PARAM(Gather)
TEST_SUITE_NAME_ONLY_LAYER_VISITOR_1_PARAM(Greater)
TEST_SUITE_NAME_ONLY_LAYER_VISITOR_1_PARAM(Maximum)
TEST_SUITE_NAME_ONLY_LAYER_VISITOR_1_PARAM(Minimum)
TEST_SUITE_NAME_ONLY_LAYER_VISITOR_1_PARAM(Multiplication)
TEST_SUITE_NAME_ONLY_LAYER_VISITOR_1_PARAM(Rsqrt)
TEST_SUITE_NAME_ONLY_LAYER_VISITOR_2_PARAM(Slice)
TEST_SUITE_NAME_ONLY_LAYER_VISITOR_1_PARAM(Subtraction)
BOOST_AUTO_TEST_SUITE_END()
|