1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
/*
* Copyright (c) 2018 Samsung Electronics Co., Ltd. All Rights Reserved
* Copyright (C) 2017 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef __NNFW_RT_TYPES_H__
#define __NNFW_RT_TYPES_H__
#include "compatibility.h"
namespace nnfw {
namespace rt {
enum class FusedActivationFunctionType { kNone, kRelu6, kRelu1, kRelu };
template <int N>
struct Dims {
int sizes[N];
int strides[N];
};
struct Shape;
inline Dims<4> convertShapeToDims(const Shape& shape) {
Dims<4> dims;
for (int i=0; i<4; i++) {
dims.sizes[i] = 1;
}
if (shape.dimensions.size() == 1) {
dims.sizes[0] = (int)getSizeOfDimension(shape, 0);
} else {
for (int i=0; i<4; i++) {
int src = (int)shape.dimensions.size()-i-1;
if (src >= 0) {
dims.sizes[i] = (int)getSizeOfDimension(shape, src);
}
}
}
dims.strides[0] = 1;
for (int i = 1; i<4; i++) {
dims.strides[i] = dims.strides[i-1] * dims.sizes[i-1];
}
return dims;
}
inline int Offset(const Dims<4>& dims, int i0, int i1, int i2, int i3) {
DCHECK(i0 >= 0 && i0 < dims.sizes[0]);
DCHECK(i1 >= 0 && i1 < dims.sizes[1]);
DCHECK(i2 >= 0 && i2 < dims.sizes[2]);
DCHECK(i3 >= 0 && i3 < dims.sizes[3]);
return i0 * dims.strides[0] + i1 * dims.strides[1] + i2 * dims.strides[2] +
i3 * dims.strides[3];
}
// Get array size, DCHECKing that the dim index is in range.
template <int N>
int ArraySize(const Dims<N>& array, int index) {
DCHECK(index >= 0 && index < N);
return array.sizes[index];
}
// Get common array size, DCHECKing that they all agree.
template <typename ArrayType1, typename ArrayType2>
int MatchingArraySize(const ArrayType1& array1, int index1,
const ArrayType2& array2, int index2) {
DCHECK_EQ(ArraySize(array1, index1), ArraySize(array2, index2));
return ArraySize(array1, index1);
}
template <typename ArrayType1, typename ArrayType2, typename... Args>
int MatchingArraySize(const ArrayType1& array1, int index1,
const ArrayType2& array2, int index2, Args... args) {
DCHECK_EQ(ArraySize(array1, index1), ArraySize(array2, index2));
return MatchingArraySize(array1, index1, args...);
}
inline int RequiredBufferSizeForDims(const Dims<4>& dims) {
int max_offset = 0;
for (int i = 0; i < 4; i++) {
max_offset += (dims.sizes[i] - 1) * dims.strides[i];
}
return max_offset + 1;
}
template <int N>
bool IsPackedWithoutStrides(const Dims<N>& dims) {
int expected_stride = 1;
for (int d = 0; d < N; d++) {
if (dims.strides[d] != expected_stride) return false;
expected_stride *= dims.sizes[d];
}
return true;
}
} // namespace rt
} // namespace nnfw
#endif // __NNFW_RT_TYPES_H__
|