modules: cmsis-nn: add pad

Add KConfig for the new cmsis-nn pad kernels.

Signed-off-by: Ryan McClelland <ryanmcclelland@meta.com>
This commit is contained in:
Ryan McClelland 2025-02-10 21:32:05 -08:00 committed by Benjamin Cabé
parent eeaf054a48
commit a8a9f0c1a9
3 changed files with 11 additions and 0 deletions

View File

@ -90,4 +90,9 @@ if(CONFIG_CMSIS_NN)
zephyr_library_sources(${SRC})
endif()
if(CONFIG_CMSIS_NN_PAD)
file(GLOB SRC "${CMSIS_NN_DIR}/Source/PadFunctions/*_s8.c")
zephyr_library_sources(${SRC})
endif()
endif()

View File

@ -78,6 +78,11 @@ config CMSIS_NN_LSTM
help
This option enables the NN libraries for Long Short-Term Memory.
config CMSIS_NN_PAD
bool "Pad"
help
This option enables the NN libraries for the pad layers.
config CMSIS_NN_TRANSPOSE
bool "Transpose"
help

View File

@ -29,6 +29,7 @@ config TENSORFLOW_LITE_MICRO_CMSIS_NN_KERNELS
select CMSIS_NN_SVD
select CMSIS_NN_LSTM
select CMSIS_NN_TRANSPOSE
select CMSIS_NN_PAD
help
This option adds support for CMSIS-NN optimized kernels when using TensorFlow Lite Micro.