forked from zephyrproject-rtos/zephyr
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathKconfig.cmsis_nn
65 lines (54 loc) · 1.8 KB
/
Kconfig.cmsis_nn
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
# Copyright (c) 2021, Commonwealth Scientific and Industrial Research
# Organisation (CSIRO) ABN 41 687 119 230.
# SPDX-License-Identifier: Apache-2.0
comment "CMSIS-NN Components"
config CMSIS_NN_ACTIVATION
bool "Activation"
help
This option enables the NN libraries for the activation layers
It can perform activation layers, including ReLU (Rectified
Linear Unit), sigmoid and tanh.
config CMSIS_NN_BASICMATH
bool "Basic Math for NN"
help
This option enables the NN libraries for the basic maths operations.
It adds functionality for element-wise add and multiplication functions.
config CMSIS_NN_CONCATENATION
bool "Concatenation"
help
This option enables the NN libraries for the concatenation layers.
config CMSIS_NN_CONVOLUTION
bool "Convolution"
imply CMSIS_NN_NNSUPPORT
help
Collection of convolution, depthwise convolution functions and
their variants. The convolution is implemented in 2 steps: im2col
and GEMM. GEMM is performed with CMSIS-DSP arm_mat_mult similar options.
config CMSIS_NN_FULLYCONNECTED
bool "Fully Connected"
imply CMSIS_NN_NNSUPPORT
help
Collection of fully-connected and matrix multiplication functions.
config CMSIS_NN_NNSUPPORT
bool "NN Support"
help
When off, its default behavior is all tables are included.
config CMSIS_NN_POOLING
bool "Pooling"
imply CMSIS_NN_NNSUPPORT
help
This option enables pooling layers, including max pooling
and average pooling.
config CMSIS_NN_RESHAPE
bool "Reshape"
help
This option enables the NN libraries for the reshape layers.
config CMSIS_NN_SOFTMAX
bool "Softmax"
help
This option enables the NN libraries for the softmax layers (exp2 based).
config CMSIS_NN_SVD
bool "SVD"
imply CMSIS_NN_NNSUPPORT
help
This option enabled the NN libraries for Single Value Decomposition Filter layers.