Compute Library
18.03
Main Page
Related Pages
Namespaces
Data Structures
Files
File List
Globals
LeNet5FullyConnectedLayerDataset.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_LENET5_FULLYCONNECTED_LAYER_DATASET
25
#define ARM_COMPUTE_TEST_LENET5_FULLYCONNECTED_LAYER_DATASET
26
27
#include "
tests/datasets/FullyConnectedLayerDataset.h
"
28
29
#include "
utils/TypePrinter.h
"
30
31
#include "
arm_compute/core/TensorShape.h
"
32
#include "
arm_compute/core/Types.h
"
33
34
using namespace
arm_compute
;
35
36
namespace
arm_compute
37
{
38
namespace
test
39
{
40
namespace
datasets
41
{
42
class
LeNet5FullyConnectedLayerDataset
final :
public
FullyConnectedLayerDataset
43
{
44
public
:
45
LeNet5FullyConnectedLayerDataset
()
46
{
47
add_config
(
TensorShape
(4
U
, 4
U
, 50
U
),
TensorShape
(800
U
, 500
U
),
TensorShape
(500
U
),
TensorShape
(500
U
));
48
add_config
(
TensorShape
(500
U
),
TensorShape
(500
U
, 10
U
),
TensorShape
(10
U
),
TensorShape
(10
U
));
49
}
50
};
51
}
// namespace datasets
52
}
// namespace test
53
}
// namespace arm_compute
54
#endif
/* ARM_COMPUTE_TEST_LENET5_FULLYCONNECTED_LAYER_DATASET */
arm_compute::test::datasets::LeNet5FullyConnectedLayerDataset
Definition:
LeNet5FullyConnectedLayerDataset.h:42
arm_compute::TensorShape
Shape of a tensor.
Definition:
TensorShape.h:39
arm_compute::test::datasets::FullyConnectedLayerDataset
Definition:
FullyConnectedLayerDataset.h:38
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
FullyConnectedLayerDataset.h
Types.h
arm_compute::test::datasets::FullyConnectedLayerDataset::add_config
void add_config(TensorShape src, TensorShape weights, TensorShape biases, TensorShape dst)
Definition:
FullyConnectedLayerDataset.h:98
TensorShape.h
arm_compute::test::datasets::LeNet5FullyConnectedLayerDataset::LeNet5FullyConnectedLayerDataset
LeNet5FullyConnectedLayerDataset()
Definition:
LeNet5FullyConnectedLayerDataset.h:45
tests
datasets
system_tests
lenet5
LeNet5FullyConnectedLayerDataset.h
Generated on Fri Mar 2 2018 12:37:54 for Compute Library by
1.8.11