24.02.1
Types.h
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2020-2023 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 ACL_ARM_COMPUTE_CORE_EXPERIMENTAL_TYPES_H
25
#define ACL_ARM_COMPUTE_CORE_EXPERIMENTAL_TYPES_H
26
27
#include "
arm_compute/core/ITensorPack.h
"
28
#include "
arm_compute/core/TensorShape.h
"
29
30
#include <vector>
31
32
namespace
arm_compute
33
{
34
// Forward declaration
35
class
ITensor;
36
37
/** Memory type */
38
enum
TensorType
: int32_t
39
{
40
ACL_UNKNOWN
= -1,
41
ACL_SRC_DST
= 0,
42
43
// Src
44
ACL_SRC
= 0,
45
ACL_SRC_0
= 0,
46
ACL_SRC_1
= 1,
47
ACL_SRC_2
= 2,
48
ACL_SRC_3
= 3,
49
ACL_SRC_4
= 4,
50
ACL_SRC_5
= 5,
51
ACL_SRC_6
= 6,
52
ACL_SRC_END
= 6,
53
54
// Dst
55
ACL_DST
= 30,
56
ACL_DST_0
= 30,
57
ACL_DST_1
= 31,
58
ACL_DST_2
= 32,
59
ACL_DST_END
= 32,
60
61
// Aux
62
ACL_INT
= 50,
63
ACL_INT_0
= 50,
64
ACL_INT_1
= 51,
65
ACL_INT_2
= 52,
66
ACL_INT_3
= 53,
67
ACL_INT_4
= 54,
68
ACL_SRC_VEC
= 256,
69
ACL_DST_VEC
= 512,
70
ACL_INT_VEC
= 1024,
71
72
// Aliasing Types
73
// Conv etc
74
ACL_BIAS
=
ACL_SRC_2
,
75
76
// Gemm
77
ACL_VEC_ROW_SUM
=
ACL_SRC_3
,
78
ACL_VEC_COL_SUM
=
ACL_SRC_4
,
79
ACL_SHIFTS
=
ACL_SRC_5
,
80
ACL_MULTIPLIERS
=
ACL_SRC_6
,
81
};
82
83
namespace
experimental
84
{
85
enum class
MemoryLifetime
86
{
87
Temporary
= 0,
88
Persistent
= 1,
89
Prepare
= 2,
90
};
91
struct
MemoryInfo
92
{
93
MemoryInfo
() =
default
;
94
95
MemoryInfo
(
int
slot
,
size_t
size
,
size_t
alignment
= 0) noexcept :
slot
(
slot
),
size
(
size
),
alignment
(
alignment
)
96
{
97
}
98
99
MemoryInfo
(
int
slot
,
MemoryLifetime
lifetime
,
size_t
size
,
size_t
alignment
= 0) noexcept
100
:
slot
(
slot
),
lifetime
(
lifetime
),
size
(
size
),
alignment
(
alignment
)
101
{
102
}
103
104
bool
merge
(
int
slot
,
size_t
new_size,
size_t
new_alignment = 0) noexcept
105
{
106
if
(
slot
!= this->slot)
107
{
108
return
false
;
109
}
110
111
size
= std::max(
size
, new_size);
112
alignment
= std::max(
alignment
, new_alignment);
113
114
return
true
;
115
}
116
117
int
slot
{
ACL_UNKNOWN
};
118
MemoryLifetime
lifetime
{
MemoryLifetime::Temporary
};
119
size_t
size
{0};
120
size_t
alignment
{64};
121
};
122
123
using
MemoryRequirements
= std::vector<MemoryInfo>;
124
}
// namespace experimental
125
}
// namespace arm_compute
126
#endif // ACL_ARM_COMPUTE_CORE_EXPERIMENTAL_TYPES_H
arm_compute::ACL_SRC_DST
@ ACL_SRC_DST
Definition:
Types.h:41
ITensorPack.h
arm_compute::experimental::MemoryRequirements
std::vector< MemoryInfo > MemoryRequirements
Definition:
Types.h:123
arm_compute::ACL_DST_1
@ ACL_DST_1
Definition:
Types.h:57
arm_compute::ACL_DST_END
@ ACL_DST_END
Definition:
Types.h:59
arm_compute::experimental::MemoryInfo::alignment
size_t alignment
Definition:
Types.h:120
arm_compute::ACL_INT_0
@ ACL_INT_0
Definition:
Types.h:63
arm_compute::ACL_UNKNOWN
@ ACL_UNKNOWN
Definition:
Types.h:40
arm_compute::experimental::MemoryInfo::MemoryInfo
MemoryInfo(int slot, size_t size, size_t alignment=0) noexcept
Definition:
Types.h:95
arm_compute::ACL_INT_2
@ ACL_INT_2
Definition:
Types.h:65
arm_compute::ACL_SHIFTS
@ ACL_SHIFTS
Definition:
Types.h:79
arm_compute::TensorType
TensorType
Memory type.
Definition:
Types.h:38
arm_compute::experimental::MemoryLifetime::Prepare
@ Prepare
arm_compute::ACL_SRC_3
@ ACL_SRC_3
Definition:
Types.h:48
arm_compute::ACL_SRC_0
@ ACL_SRC_0
Definition:
Types.h:45
arm_compute::ACL_SRC_6
@ ACL_SRC_6
Definition:
Types.h:51
arm_compute::ACL_SRC_END
@ ACL_SRC_END
Definition:
Types.h:52
arm_compute::ACL_SRC_1
@ ACL_SRC_1
Definition:
Types.h:46
arm_compute::ACL_INT_4
@ ACL_INT_4
Definition:
Types.h:67
arm_compute::ACL_SRC_2
@ ACL_SRC_2
Definition:
Types.h:47
arm_compute::ACL_DST_2
@ ACL_DST_2
Definition:
Types.h:58
arm_compute::experimental::MemoryInfo
Definition:
Types.h:91
arm_compute::ACL_SRC_5
@ ACL_SRC_5
Definition:
Types.h:50
arm_compute::experimental::MemoryInfo::lifetime
MemoryLifetime lifetime
Definition:
Types.h:118
arm_compute::experimental::MemoryLifetime
MemoryLifetime
Definition:
Types.h:85
arm_compute::ACL_DST_0
@ ACL_DST_0
Definition:
Types.h:56
arm_compute::ACL_INT_1
@ ACL_INT_1
Definition:
Types.h:64
arm_compute::ACL_SRC_4
@ ACL_SRC_4
Definition:
Types.h:49
arm_compute::ACL_DST
@ ACL_DST
Definition:
Types.h:55
arm_compute::experimental::MemoryInfo::merge
bool merge(int slot, size_t new_size, size_t new_alignment=0) noexcept
Definition:
Types.h:104
arm_compute::ACL_MULTIPLIERS
@ ACL_MULTIPLIERS
Definition:
Types.h:80
arm_compute::experimental::MemoryLifetime::Temporary
@ Temporary
arm_compute::ACL_SRC_VEC
@ ACL_SRC_VEC
Definition:
Types.h:68
arm_compute::experimental::MemoryInfo::MemoryInfo
MemoryInfo(int slot, MemoryLifetime lifetime, size_t size, size_t alignment=0) noexcept
Definition:
Types.h:99
arm_compute::ACL_INT
@ ACL_INT
Definition:
Types.h:62
arm_compute::experimental::MemoryLifetime::Persistent
@ Persistent
arm_compute::ACL_INT_3
@ ACL_INT_3
Definition:
Types.h:66
arm_compute::ACL_BIAS
@ ACL_BIAS
Definition:
Types.h:74
arm_compute
Copyright (c) 2017-2024 Arm Limited.
Definition:
introduction.dox:24
arm_compute::ACL_VEC_COL_SUM
@ ACL_VEC_COL_SUM
Definition:
Types.h:78
TensorShape.h
arm_compute::experimental::MemoryInfo::MemoryInfo
MemoryInfo()=default
arm_compute::experimental::MemoryInfo::slot
int slot
Definition:
Types.h:117
arm_compute::ACL_INT_VEC
@ ACL_INT_VEC
Definition:
Types.h:70
arm_compute::ACL_SRC
@ ACL_SRC
Definition:
Types.h:44
arm_compute::experimental::MemoryInfo::size
size_t size
Definition:
Types.h:119
arm_compute::ACL_DST_VEC
@ ACL_DST_VEC
Definition:
Types.h:69
arm_compute::ACL_VEC_ROW_SUM
@ ACL_VEC_ROW_SUM
Definition:
Types.h:77
arm_compute
core
experimental
Types.h
Generated on Mon Mar 18 2024 11:31:54 for Compute Library by
1.8.17