|
| 1 | +# Copyright (c) 2021, Commonwealth Scientific and Industrial Research |
| 2 | +# Organisation (CSIRO) ABN 41 687 119 230. |
| 3 | +# SPDX-License-Identifier: Apache-2.0 |
| 4 | + |
| 5 | +comment "CMSIS-NN Components" |
| 6 | + |
| 7 | +config CMSIS_NN_ACTIVATION |
| 8 | + bool "Activation" |
| 9 | + help |
| 10 | + This option enables the NN libraries for the activation layers |
| 11 | + It can perform activation layers, including ReLU (Rectified |
| 12 | + Linear Unit), sigmoid and tanh. |
| 13 | + |
| 14 | +config CMSIS_NN_BASICMATH |
| 15 | + bool "Basic Math for NN" |
| 16 | + help |
| 17 | + This option enables the NN libraries for the basic maths operations. |
| 18 | + It adds functionality for element-wise add and multiplication functions. |
| 19 | + |
| 20 | +config CMSIS_NN_CONCATENATION |
| 21 | + bool "Concatenation" |
| 22 | + help |
| 23 | + This option enables the NN libraries for the concatenation layers. |
| 24 | + |
| 25 | +config CMSIS_NN_CONVOLUTION |
| 26 | + bool "Convolution" |
| 27 | + imply CMSIS_NN_NNSUPPORT |
| 28 | + help |
| 29 | + Collection of convolution, depthwise convolution functions and |
| 30 | + their variants. The convolution is implemented in 2 steps: im2col |
| 31 | + and GEMM. GEMM is performed with CMSIS-DSP arm_mat_mult similar options. |
| 32 | + |
| 33 | +config CMSIS_NN_FULLYCONNECTED |
| 34 | + bool "Fully Connected" |
| 35 | + imply CMSIS_NN_NNSUPPORT |
| 36 | + help |
| 37 | + Collection of fully-connected and matrix multiplication functions. |
| 38 | + |
| 39 | +config CMSIS_NN_NNSUPPORT |
| 40 | + bool "NN Support" |
| 41 | + help |
| 42 | + When off, its default behavior is all tables are included. |
| 43 | + |
| 44 | +config CMSIS_NN_POOLING |
| 45 | + bool "Pooling" |
| 46 | + imply CMSIS_NN_NNSUPPORT |
| 47 | + help |
| 48 | + This option enables pooling layers, including max pooling |
| 49 | + and average pooling. |
| 50 | + |
| 51 | +config CMSIS_NN_RESHAPE |
| 52 | + bool "Reshape" |
| 53 | + help |
| 54 | + This option enables the NN libraries for the reshape layers. |
| 55 | + |
| 56 | +config CMSIS_NN_SOFTMAX |
| 57 | + bool "Softmax" |
| 58 | + help |
| 59 | + This option enables the NN libraries for the softmax layers (exp2 based). |
| 60 | + |
| 61 | +config CMSIS_NN_SVD |
| 62 | + bool "SVD" |
| 63 | + imply CMSIS_NN_NNSUPPORT |
| 64 | + help |
| 65 | + This option enabled the NN libraries for Single Value Decomposition Filter layers. |
0 commit comments