aboutsummaryrefslogtreecommitdiff
path: root/tflite_micro.cmake
diff options
context:
space:
mode:
Diffstat (limited to 'tflite_micro.cmake')
-rw-r--r--tflite_micro.cmake4
1 files changed, 3 insertions, 1 deletions
diff --git a/tflite_micro.cmake b/tflite_micro.cmake
index 2d5bc3b..ef30fa8 100644
--- a/tflite_micro.cmake
+++ b/tflite_micro.cmake
@@ -186,11 +186,13 @@ tensorflow_target_sources_glob(tflu GLOB TRUE
${TFLU_PATH}/*.cc
${TFLU_PATH}/arena_allocator/*.cc
${TFLU_PATH}/memory_planner/*.cc
- ${TFLU_PATH}/kernels/*.cc)
+ ${TFLU_PATH}/kernels/*.cc
+ ${TFLU_PATH}/tflite_bridge/*.cc)
tensorflow_target_sources_glob(tflu GLOB_RECURSE FALSE
${TFLU_PATH}/../c/*.cc
${TFLU_PATH}/../core/*.cc
+ ${TFLU_PATH}/../core/api/*.cc
${TFLU_PATH}/../kernels/*.cc
${TFLU_PATH}/../schema/*.cc)