ArmNN
NotReleased
InternalTypes.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
#pragma once
6
7
#include <
armnn/Types.hpp
>
8
9
#include <array>
10
11
namespace
armnn
12
{
13
14
enum class
LayerType
15
{
16
FirstLayer
,
17
Activation
=
FirstLayer
,
18
Addition
,
19
ArgMinMax
,
20
BatchNormalization
,
21
BatchToSpaceNd
,
22
Comparison
,
23
Concat
,
24
Constant
,
25
ConvertFp16ToFp32
,
26
ConvertFp32ToFp16
,
27
Convolution2d
,
28
Debug
,
29
DepthToSpace
,
30
DepthwiseConvolution2d
,
31
Dequantize
,
32
DetectionPostProcess
,
33
Division
,
34
ElementwiseUnary
,
35
FakeQuantization
,
36
Floor
,
37
FullyConnected
,
38
Gather
,
39
Input
,
40
InstanceNormalization
,
41
L2Normalization
,
42
LogSoftmax
,
43
Lstm
,
44
Maximum
,
45
Mean
,
46
MemCopy
,
47
MemImport
,
48
Merge
,
49
Minimum
,
50
Multiplication
,
51
Normalization
,
52
Output
,
53
Pad
,
54
Permute
,
55
Pooling2d
,
56
PreCompiled
,
57
Prelu
,
58
Quantize
,
59
QuantizedLstm
,
60
Reshape
,
61
Resize
,
62
Slice
,
63
Softmax
,
64
SpaceToBatchNd
,
65
SpaceToDepth
,
66
Splitter
,
67
Stack
,
68
StandIn
,
69
StridedSlice
,
70
Subtraction
,
71
Switch
,
72
// Last layer goes here.
73
LastLayer
,
74
TransposeConvolution2d
= LastLayer
75
};
76
77
const
char
*
GetLayerTypeAsCString
(
LayerType
type);
78
79
using
Coordinates
= std::array<unsigned int, MaxNumOfTensorDimensions>;
80
using
Dimensions
= std::array<unsigned int, MaxNumOfTensorDimensions>;
81
82
}
armnn::LayerType::ElementwiseUnary
armnn::LayerType::Addition
armnn::LayerType::Multiplication
armnn::LayerType::FakeQuantization
armnn::LayerType::Comparison
armnn::LayerType::ConvertFp16ToFp32
armnn::LayerType::Merge
armnn::GetLayerTypeAsCString
char const * GetLayerTypeAsCString(LayerType type)
Definition:
InternalTypes.cpp:13
armnn::LayerType::Slice
armnn::LayerType::DepthToSpace
armnn::LayerType::SpaceToBatchNd
armnn::LayerType::QuantizedLstm
armnn::LayerType::Output
armnn::LayerType::Splitter
armnn::LayerType::L2Normalization
armnn::LayerType::StandIn
armnn::LayerType::Activation
armnn::LayerType::FullyConnected
armnn::LayerType::Constant
armnn::LayerType::Stack
armnn::LayerType::Maximum
armnn::Dimensions
std::array< unsigned int, MaxNumOfTensorDimensions > Dimensions
Definition:
InternalTypes.hpp:80
armnn::LayerType::BatchToSpaceNd
armnn::LayerType::InstanceNormalization
armnn::Coordinates
std::array< unsigned int, MaxNumOfTensorDimensions > Coordinates
Definition:
InternalTypes.hpp:79
armnn::LayerType::MemImport
armnn::LayerType::Normalization
armnn::LayerType::Reshape
armnn::LayerType::Softmax
armnn::LayerType::Quantize
armnn::LogSeverity::Debug
armnn::LayerType::Prelu
armnn
Definition:
BackendHelper.hpp:11
Types.hpp
armnn::LayerType::DetectionPostProcess
armnn::LayerType::Resize
armnn::LayerType::ArgMinMax
armnn::LayerType::Minimum
armnn::LayerType::ConvertFp32ToFp16
armnn::LayerType::Convolution2d
armnn::LayerType::BatchNormalization
armnn::LayerType::MemCopy
armnn::LayerType::FirstLayer
armnn::LayerType::Mean
armnn::LayerType::Lstm
armnn::LayerType::Pooling2d
armnn::LayerType::Pad
armnn::OutputShapeRounding::Floor
armnn::LayerType::SpaceToDepth
armnn::LayerType::Permute
armnn::LayerType::StridedSlice
armnn::LayerType::Division
armnn::LayerType::LastLayer
armnn::LayerType::Gather
armnn::LayerType::DepthwiseConvolution2d
armnn::LayerType::Subtraction
armnn::LayerType::Input
armnn::LayerType::Dequantize
armnn::LayerType::LogSoftmax
armnn::LayerType::Concat
armnn::LayerType::Switch
armnn::LayerType
LayerType
Definition:
InternalTypes.hpp:14
armnn::LayerType::TransposeConvolution2d
armnn::LayerType::PreCompiled
src
armnn
InternalTypes.hpp
Generated on Fri Mar 13 2020 16:06:54 for ArmNN by
1.8.13