Compute Library
18.03
Main Page
Related Pages
Namespaces
Data Structures
Files
File List
Globals
SqueezeNetConvolutionLayerDataset.h
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2017 ARM Limited.
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
#ifndef ARM_COMPUTE_TEST_SQUEEZENET_CONVOLUTION_LAYER_DATASET
25
#define ARM_COMPUTE_TEST_SQUEEZENET_CONVOLUTION_LAYER_DATASET
26
27
#include "
tests/datasets/ConvolutionLayerDataset.h
"
28
29
#include "
utils/TypePrinter.h
"
30
31
#include "
arm_compute/core/TensorShape.h
"
32
#include "
arm_compute/core/Types.h
"
33
34
namespace
arm_compute
35
{
36
namespace
test
37
{
38
namespace
datasets
39
{
40
class
SqueezeNetWinogradLayerDataset
final :
public
ConvolutionLayerDataset
41
{
42
public
:
43
SqueezeNetWinogradLayerDataset
()
44
{
45
// fire2/expand3x3, fire3/expand3x3
46
add_config
(
TensorShape
(55
U
, 55
U
, 16
U
),
TensorShape
(3
U
, 3
U
, 16
U
, 64
U
),
TensorShape
(64
U
),
TensorShape
(55
U
, 55
U
, 64
U
),
PadStrideInfo
(1, 1, 1, 1));
47
// fire4/expand3x3, fire5/expand3x3
48
add_config
(
TensorShape
(27
U
, 27
U
, 32
U
),
TensorShape
(3
U
, 3
U
, 32
U
, 128
U
),
TensorShape
(128
U
),
TensorShape
(27
U
, 27
U
, 128
U
),
PadStrideInfo
(1, 1, 1, 1));
49
// fire6/expand3x3, fire7/expand3x3
50
add_config
(
TensorShape
(13
U
, 13
U
, 48
U
),
TensorShape
(3
U
, 3
U
, 48
U
, 192
U
),
TensorShape
(192
U
),
TensorShape
(13
U
, 13
U
, 192
U
),
PadStrideInfo
(1, 1, 1, 1));
51
// fire8/expand3x3, fire9/expand3x3
52
add_config
(
TensorShape
(13
U
, 13
U
, 64
U
),
TensorShape
(3
U
, 3
U
, 64
U
, 256
U
),
TensorShape
(256
U
),
TensorShape
(13
U
, 13
U
, 256
U
),
PadStrideInfo
(1, 1, 1, 1));
53
}
54
};
55
56
class
SqueezeNetConvolutionLayerDataset
final :
public
ConvolutionLayerDataset
57
{
58
public
:
59
SqueezeNetConvolutionLayerDataset
()
60
{
61
// conv1
62
add_config
(
TensorShape
(224
U
, 224
U
, 3
U
),
TensorShape
(3
U
, 3
U
, 3
U
, 64
U
),
TensorShape
(64
U
),
TensorShape
(111
U
, 111
U
, 64
U
),
PadStrideInfo
(2, 2, 0, 0));
63
// fire2/squeeze1x1
64
add_config
(
TensorShape
(55
U
, 55
U
, 64
U
),
TensorShape
(1
U
, 1
U
, 64
U
, 16
U
),
TensorShape
(16
U
),
TensorShape
(55
U
, 55
U
, 16
U
),
PadStrideInfo
(1, 1, 0, 0));
65
// fire2/expand1x1, fire3/expand1x1
66
add_config
(
TensorShape
(55
U
, 55
U
, 16
U
),
TensorShape
(1
U
, 1
U
, 16
U
, 64
U
),
TensorShape
(64
U
),
TensorShape
(55
U
, 55
U
, 64
U
),
PadStrideInfo
(1, 1, 0, 0));
67
// fire2/expand3x3, fire3/expand3x3
68
add_config
(
TensorShape
(55
U
, 55
U
, 16
U
),
TensorShape
(3
U
, 3
U
, 16
U
, 64
U
),
TensorShape
(64
U
),
TensorShape
(55
U
, 55
U
, 64
U
),
PadStrideInfo
(1, 1, 1, 1));
69
// fire3/squeeze1x1
70
add_config
(
TensorShape
(55
U
, 55
U
, 128
U
),
TensorShape
(1
U
, 1
U
, 128
U
, 16
U
),
TensorShape
(16
U
),
TensorShape
(55
U
, 55
U
, 16
U
),
PadStrideInfo
(1, 1, 0, 0));
71
// fire4/squeeze1x1
72
add_config
(
TensorShape
(27
U
, 27
U
, 128
U
),
TensorShape
(1
U
, 1
U
, 128
U
, 32
U
),
TensorShape
(32
U
),
TensorShape
(27
U
, 27
U
, 32
U
),
PadStrideInfo
(1, 1, 0, 0));
73
// fire4/expand1x1, fire5/expand1x1
74
add_config
(
TensorShape
(27
U
, 27
U
, 32
U
),
TensorShape
(1
U
, 1
U
, 32
U
, 128
U
),
TensorShape
(128
U
),
TensorShape
(27
U
, 27
U
, 128
U
),
PadStrideInfo
(1, 1, 0, 0));
75
// fire4/expand3x3, fire5/expand3x3
76
add_config
(
TensorShape
(27
U
, 27
U
, 32
U
),
TensorShape
(3
U
, 3
U
, 32
U
, 128
U
),
TensorShape
(128
U
),
TensorShape
(27
U
, 27
U
, 128
U
),
PadStrideInfo
(1, 1, 1, 1));
77
// fire5/squeeze1x1
78
add_config
(
TensorShape
(27
U
, 27
U
, 256
U
),
TensorShape
(1
U
, 1
U
, 256
U
, 32
U
),
TensorShape
(32
U
),
TensorShape
(27
U
, 27
U
, 32
U
),
PadStrideInfo
(1, 1, 0, 0));
79
// fire6/squeeze1x1
80
add_config
(
TensorShape
(13
U
, 13
U
, 256
U
),
TensorShape
(1
U
, 1
U
, 256
U
, 48
U
),
TensorShape
(48
U
),
TensorShape
(13
U
, 13
U
, 48
U
),
PadStrideInfo
(1, 1, 0, 0));
81
// fire6/expand1x1, fire7/expand1x1
82
add_config
(
TensorShape
(13
U
, 13
U
, 48
U
),
TensorShape
(1
U
, 1
U
, 48
U
, 192
U
),
TensorShape
(192
U
),
TensorShape
(13
U
, 13
U
, 192
U
),
PadStrideInfo
(1, 1, 0, 0));
83
// fire6/expand3x3, fire7/expand3x3
84
add_config
(
TensorShape
(13
U
, 13
U
, 48
U
),
TensorShape
(3
U
, 3
U
, 48
U
, 192
U
),
TensorShape
(192
U
),
TensorShape
(13
U
, 13
U
, 192
U
),
PadStrideInfo
(1, 1, 1, 1));
85
// fire7/squeeze1x1
86
add_config
(
TensorShape
(13
U
, 13
U
, 384
U
),
TensorShape
(1
U
, 1
U
, 384
U
, 48
U
),
TensorShape
(48
U
),
TensorShape
(13
U
, 13
U
, 48
U
),
PadStrideInfo
(1, 1, 0, 0));
87
// fire8/squeeze1x1
88
add_config
(
TensorShape
(13
U
, 13
U
, 384
U
),
TensorShape
(1
U
, 1
U
, 384
U
, 64
U
),
TensorShape
(64
U
),
TensorShape
(13
U
, 13
U
, 64
U
),
PadStrideInfo
(1, 1, 0, 0));
89
// fire8/expand1x1, fire9/expand1x1
90
add_config
(
TensorShape
(13
U
, 13
U
, 64
U
),
TensorShape
(1
U
, 1
U
, 64
U
, 256
U
),
TensorShape
(256
U
),
TensorShape
(13
U
, 13
U
, 256
U
),
PadStrideInfo
(1, 1, 0, 0));
91
// fire8/expand3x3, fire9/expand3x3
92
add_config
(
TensorShape
(13
U
, 13
U
, 64
U
),
TensorShape
(3
U
, 3
U
, 64
U
, 256
U
),
TensorShape
(256
U
),
TensorShape
(13
U
, 13
U
, 256
U
),
PadStrideInfo
(1, 1, 1, 1));
93
// fire9/squeeze1x1
94
add_config
(
TensorShape
(13
U
, 13
U
, 512
U
),
TensorShape
(1
U
, 1
U
, 512
U
, 64
U
),
TensorShape
(64
U
),
TensorShape
(13
U
, 13
U
, 64
U
),
PadStrideInfo
(1, 1, 0, 0));
95
// conv10
96
add_config
(
TensorShape
(13
U
, 13
U
, 512
U
),
TensorShape
(1
U
, 1
U
, 512
U
, 1000
U
),
TensorShape
(1000
U
),
TensorShape
(13
U
, 13
U
, 1000
U
),
PadStrideInfo
(1, 1, 0, 0));
97
}
98
};
99
}
// namespace datasets
100
}
// namespace test
101
}
// namespace arm_compute
102
#endif
/* ARM_COMPUTE_TEST_SQUEEZENET_CONVOLUTION_LAYER_DATASET */
arm_compute::test::datasets::SqueezeNetConvolutionLayerDataset
Definition:
SqueezeNetConvolutionLayerDataset.h:56
arm_compute::TensorShape
Shape of a tensor.
Definition:
TensorShape.h:39
arm_compute::test::datasets::ConvolutionLayerDataset::add_config
void add_config(TensorShape src, TensorShape weights, TensorShape biases, TensorShape dst, PadStrideInfo info)
Definition:
ConvolutionLayerDataset.h:103
arm_compute::test::datasets::SqueezeNetWinogradLayerDataset
Definition:
SqueezeNetConvolutionLayerDataset.h:40
arm_compute::test::datasets::SqueezeNetWinogradLayerDataset::SqueezeNetWinogradLayerDataset
SqueezeNetWinogradLayerDataset()
Definition:
SqueezeNetConvolutionLayerDataset.h:43
arm_compute
This file contains all available output stages for GEMMLowp on OpenCL.
Definition:
00_introduction.dox:1
arm_compute::Channel::U
Cb/U channel.
TypePrinter.h
arm_compute::test::datasets::SqueezeNetConvolutionLayerDataset::SqueezeNetConvolutionLayerDataset
SqueezeNetConvolutionLayerDataset()
Definition:
SqueezeNetConvolutionLayerDataset.h:59
arm_compute::PadStrideInfo
Padding and stride information class.
Definition:
Types.h:491
ConvolutionLayerDataset.h
arm_compute::test::datasets::ConvolutionLayerDataset
Definition:
ConvolutionLayerDataset.h:38
Types.h
TensorShape.h
tests
datasets
system_tests
squeezenet
SqueezeNetConvolutionLayerDataset.h
Generated on Fri Mar 2 2018 12:37:54 for Compute Library by
1.8.11