Skip to content

Commit a8a9f0c

Browse files
XenuIsWatchingkartben
authored andcommitted
modules: cmsis-nn: add pad
Add KConfig for the new cmsis-nn pad kernels. Signed-off-by: Ryan McClelland <ryanmcclelland@meta.com>
1 parent eeaf054 commit a8a9f0c

File tree

3 files changed

+11
-0
lines changed

3 files changed

+11
-0
lines changed

modules/cmsis-nn/CMakeLists.txt

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -90,4 +90,9 @@ if(CONFIG_CMSIS_NN)
9090
zephyr_library_sources(${SRC})
9191
endif()
9292

93+
if(CONFIG_CMSIS_NN_PAD)
94+
file(GLOB SRC "${CMSIS_NN_DIR}/Source/PadFunctions/*_s8.c")
95+
zephyr_library_sources(${SRC})
96+
endif()
97+
9398
endif()

modules/cmsis-nn/Kconfig

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,11 @@ config CMSIS_NN_LSTM
7878
help
7979
This option enables the NN libraries for Long Short-Term Memory.
8080

81+
config CMSIS_NN_PAD
82+
bool "Pad"
83+
help
84+
This option enables the NN libraries for the pad layers.
85+
8186
config CMSIS_NN_TRANSPOSE
8287
bool "Transpose"
8388
help

modules/tflite-micro/Kconfig

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ config TENSORFLOW_LITE_MICRO_CMSIS_NN_KERNELS
2929
select CMSIS_NN_SVD
3030
select CMSIS_NN_LSTM
3131
select CMSIS_NN_TRANSPOSE
32+
select CMSIS_NN_PAD
3233
help
3334
This option adds support for CMSIS-NN optimized kernels when using TensorFlow Lite Micro.
3435

0 commit comments

Comments
 (0)