Browse Source

modules: cmsis-nn: add pad

Add KConfig for the new cmsis-nn pad kernels.

Signed-off-by: Ryan McClelland <ryanmcclelland@meta.com>
pull/85708/head
Ryan McClelland 5 months ago committed by Benjamin Cabé
parent
commit
a8a9f0c1a9
  1. 5
      modules/cmsis-nn/CMakeLists.txt
  2. 5
      modules/cmsis-nn/Kconfig
  3. 1
      modules/tflite-micro/Kconfig

5
modules/cmsis-nn/CMakeLists.txt

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

5
modules/cmsis-nn/Kconfig

@ -78,6 +78,11 @@ config CMSIS_NN_LSTM
help help
This option enables the NN libraries for Long Short-Term Memory. 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 config CMSIS_NN_TRANSPOSE
bool "Transpose" bool "Transpose"
help help

1
modules/tflite-micro/Kconfig

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

Loading…
Cancel
Save