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:
parent
eeaf054a48
commit
a8a9f0c1a9
@ -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()
|
||||
|
||||
@ -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
|
||||
|
||||
@ -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.
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user