Blame - tests/benchmark/NEON/BatchNormalizationLayer.cpp - ml/ComputeLibrary

SiCong Li

1e5c157

2017-07-27 17:58:52 +0100

[diff] [blame]

1

/*

Georgios Pinitas

51a6730

2018-01-17 12:55:26 +0000

[diff] [blame]

2

SiCong Li

1e5c157

2017-07-27 17:58:52 +0100

[diff] [blame]

3

*

4

* SPDX-License-Identifier: MIT

5

*

6

* Permission is hereby granted, free of charge, to any person obtaining a copy

7

* of this software and associated documentation files (the "Software"), to

8

* deal in the Software without restriction, including without limitation the

9

* rights to use, copy, modify, merge, publish, distribute, sublicense, and/or

10

* sell copies of the Software, and to permit persons to whom the Software is

11

* furnished to do so, subject to the following conditions:

12

*

13

* The above copyright notice and this permission notice shall be included in all

14

* copies or substantial portions of the Software.

15

*

16

* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

17

* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

18

* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

19

* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

20

* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

21

* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE

22

* SOFTWARE.

23

*/

24

#include "arm_compute/core/TensorShape.h"

25

#include "arm_compute/core/Types.h"

26

#include "arm_compute/runtime/NEON/functions/NEBatchNormalizationLayer.h"

27

#include "arm_compute/runtime/Tensor.h"

28

#include "arm_compute/runtime/TensorAllocator.h"

SiCong Li

1e5c157

2017-07-27 17:58:52 +0100

[diff] [blame]

29

#include "tests/NEON/Accessor.h"

Moritz Pflanzer

a09de0c

2017-09-01 20:41:12 +0100

[diff] [blame]

30

#include "tests/framework/Macros.h"

31

#include "tests/framework/datasets/Datasets.h"

Anthony Barbier

2a07e18

2017-08-04 18:20:27 +0100

[diff] [blame]

32

#include "utils/TypePrinter.h"

SiCong Li

1e5c157

2017-07-27 17:58:52 +0100

[diff] [blame]

33

Moritz Pflanzer

a09de0c

2017-09-01 20:41:12 +0100

[diff] [blame]

34

#include "tests/benchmark/fixtures/BatchNormalizationLayerFixture.h"

35

#include "tests/datasets/system_tests/googlenet/inceptionv4/GoogLeNetInceptionV4BatchNormalizationLayerDataset.h"

Giorgio Arena

1167487

2018-02-07 15:38:12 +0000

[diff] [blame]

36

#include "tests/datasets/system_tests/mobilenet/MobileNetBatchNormalizationLayerDataset.h"

Moritz Pflanzer

a09de0c

2017-09-01 20:41:12 +0100

[diff] [blame]

37

#include "tests/datasets/system_tests/yolo/v2/YOLOV2BatchNormalizationLayerDataset.h"

SiCong Li

1e5c157

2017-07-27 17:58:52 +0100

[diff] [blame]

38

39

namespace arm_compute

40

{

41

namespace test

42

{

Michalis Spyrou

724079d

2017-12-15 11:37:37 +0000

[diff] [blame]

43

namespace benchmark

44

{

SiCong Li

1e5c157

2017-07-27 17:58:52 +0100

[diff] [blame]

45

namespace

46

{

Ioan-Cristian Szabo

5edbd1c

2017-11-13 13:34:08 +0000

[diff] [blame]

47

#ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC

Georgios Pinitas

51a6730

2018-01-17 12:55:26 +0000

[diff] [blame]

48

const auto data_types = framework::dataset::make("DataType", { DataType::F16, DataType::F32 });

Ioan-Cristian Szabo

5edbd1c

2017-11-13 13:34:08 +0000

[diff] [blame]

49

#else /* __ARM_FEATURE_FP16_VECTOR_ARITHMETIC */

Georgios Pinitas

51a6730

2018-01-17 12:55:26 +0000

[diff] [blame]

50

const auto data_types = framework::dataset::make("DataType", { DataType::F32 });

Ioan-Cristian Szabo

5edbd1c

2017-11-13 13:34:08 +0000

[diff] [blame]

51

#endif /* __ARM_FEATURE_FP16_VECTOR_ARITHMETIC */

SiCong Li

1e5c157

2017-07-27 17:58:52 +0100

[diff] [blame]

52

} // namespace

53

54

using NEBatchNormalizationLayerFixture = BatchNormalizationLayerFixture<Tensor, NEBatchNormalizationLayer, Accessor>;

TEST_SUITE(NEON)

Giorgio Arena

2018-02-07 15:38:12 +0000

[diff] [blame]

58

REGISTER_FIXTURE_DATA_TEST_CASE(MobileNetBatchNormalizationLayer, NEBatchNormalizationLayerFixture, framework::DatasetMode::ALL,

Michele Di Giorgio

0cbb927

2018-03-01 16:56:48 +0000

[diff] [blame]

59

combine(combine(combine(combine(combine(datasets::MobileNetBatchNormalizationLayerDataset(),

60

combine(framework::dataset::make("UseGamma", { false, true }),

61

framework::dataset::make("UseBeta", { false, true }))),

62

framework::dataset::make("ActivationInfo", ActivationLayerInfo())),

63

data_types),

64

framework::dataset::make("DataLayout", { DataLayout::NCHW })),

65

framework::dataset::make("Batches", 1)));

SiCong Li

1e5c157

2017-07-27 17:58:52 +0100

[diff] [blame]

66

REGISTER_FIXTURE_DATA_TEST_CASE(YOLOV2BatchNormalizationLayer, NEBatchNormalizationLayerFixture, framework::DatasetMode::ALL,

Michele Di Giorgio

0cbb927

2018-03-01 16:56:48 +0000

[diff] [blame]

67

combine(combine(combine(combine(combine(datasets::YOLOV2BatchNormalizationLayerDataset(),

68

combine(framework::dataset::make("UseGamma", { false, true }),

69

framework::dataset::make("UseBeta", { false, true }))),

70

framework::dataset::make("ActivationInfo", ActivationLayerInfo())),

71

data_types),

72

framework::dataset::make("DataLayout", { DataLayout::NCHW })),

73

framework::dataset::make("Batches", 1)));

SiCong Li

1e5c157

2017-07-27 17:58:52 +0100

[diff] [blame]

74

SiCong Li

f583fb7

2017-08-15 15:09:18 +0100

[diff] [blame]

75

REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV4BatchNormalizationLayer, NEBatchNormalizationLayerFixture, framework::DatasetMode::ALL,

Michele Di Giorgio

0cbb927

2018-03-01 16:56:48 +0000

[diff] [blame]

76

combine(combine(combine(combine(combine(datasets::GoogLeNetInceptionV4BatchNormalizationLayerDataset(),

77

combine(framework::dataset::make("UseGamma", { false, true }),

78

framework::dataset::make("UseBeta", { false, true }))),

79

framework::dataset::make("ActivationInfo", ActivationLayerInfo())),

80

data_types),

81

framework::dataset::make("DataLayout", { DataLayout::NCHW })),

82

framework::dataset::make("Batches", 1)));

SiCong Li

f583fb7

2017-08-15 15:09:18 +0100

[diff] [blame]

83

SiCong Li

1e5c157

2017-07-27 17:58:52 +0100

[diff] [blame]

84

TEST_SUITE(NIGHTLY)

Giorgio Arena

1167487

2018-02-07 15:38:12 +0000

[diff] [blame]

85

REGISTER_FIXTURE_DATA_TEST_CASE(MobileNetBatchNormalizationLayer, NEBatchNormalizationLayerFixture, framework::DatasetMode::NIGHTLY,

Michele Di Giorgio

0cbb927

2018-03-01 16:56:48 +0000

[diff] [blame]

86

combine(combine(combine(combine(combine(datasets::MobileNetBatchNormalizationLayerDataset(),

87

combine(framework::dataset::make("UseGamma", { false, true }),

88

framework::dataset::make("UseBeta", { false, true }))),

89

framework::dataset::make("ActivationInfo", ActivationLayerInfo())),

90

data_types),

91

framework::dataset::make("DataLayout", { DataLayout::NCHW })),

92

framework::dataset::make("Batches", { 4, 8 })));

SiCong Li

1e5c157

2017-07-27 17:58:52 +0100

[diff] [blame]

93

REGISTER_FIXTURE_DATA_TEST_CASE(YOLOV2BatchNormalizationLayer, NEBatchNormalizationLayerFixture, framework::DatasetMode::NIGHTLY,

Michele Di Giorgio

0cbb927

2018-03-01 16:56:48 +0000

[diff] [blame]

94

combine(combine(combine(combine(combine(datasets::YOLOV2BatchNormalizationLayerDataset(),

95

combine(framework::dataset::make("UseGamma", { false, true }),

96

framework::dataset::make("UseBeta", { false, true }))),

97

framework::dataset::make("ActivationInfo", ActivationLayerInfo())),

98

data_types),

99

framework::dataset::make("DataLayout", { DataLayout::NCHW })),

100

framework::dataset::make("Batches", { 4, 8 })));

SiCong Li

f583fb7

2017-08-15 15:09:18 +0100

[diff] [blame]

101

102

REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV4BatchNormalizationLayer, NEBatchNormalizationLayerFixture, framework::DatasetMode::NIGHTLY,

Michele Di Giorgio

0cbb927

2018-03-01 16:56:48 +0000

[diff] [blame]

103

combine(combine(combine(combine(combine(datasets::GoogLeNetInceptionV4BatchNormalizationLayerDataset(),

104

combine(framework::dataset::make("UseGamma", { false, true }),

105

framework::dataset::make("UseBeta", { false, true }))),

106

framework::dataset::make("ActivationInfo", ActivationLayerInfo())),

107

data_types),

108

framework::dataset::make("DataLayout", { DataLayout::NCHW })),

109

framework::dataset::make("Batches", { 4, 8 })));

SiCong Li

1e5c157

2017-07-27 17:58:52 +0100

[diff] [blame]

110

TEST_SUITE_END()

111

TEST_SUITE_END()

Michalis Spyrou

724079d

2017-12-15 11:37:37 +0000

[diff] [blame]

112

} // namespace benchmark

SiCong Li

1e5c157

2017-07-27 17:58:52 +0100

[diff] [blame]

113

} // namespace test

114

} // namespace arm_compute