From a5e90fd528f425a32a1d26a3d794be0eb4e0b5f8 Mon Sep 17 00:00:00 2001 From: Jens Elofsson Date: Tue, 1 Jun 2021 19:06:30 +0200 Subject: Switch to the new tflite-micro repo. Change-Id: I19b806e94207580ab548e8cd4ddada32debf0639 --- CMakeLists.txt | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'CMakeLists.txt') diff --git a/CMakeLists.txt b/CMakeLists.txt index 7f5f1cb..cc0d39e 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -31,7 +31,7 @@ set(CMSIS_PATH "${CMAKE_CURRENT_SOURCE_DIR}/cmsis" CACHE PATH "Path to CMSIS.") set(CMSIS_VIEW_PATH "${CMAKE_CURRENT_SOURCE_DIR}/cmsis-view" CACHE PATH "Path to cmsis-view.") set(CORE_DRIVER_PATH "${CMAKE_CURRENT_SOURCE_DIR}/core_driver" CACHE PATH "Path to core driver.") set(LINUX_DRIVER_STACK_PATH "${CMAKE_CURRENT_SOURCE_DIR}/../linux_driver_stack" CACHE PATH "Path to Linux driver stack for Arm Ethos-U.") -set(TENSORFLOW_PATH "${CMAKE_CURRENT_SOURCE_DIR}/tensorflow" CACHE PATH "Path to Tensorflow.") +set(TENSORFLOW_PATH "${CMAKE_CURRENT_SOURCE_DIR}/tflite_micro" CACHE PATH "Path to Tensorflow Lite Micro.") set(TFLU_PREBUILT_LIBRARY_PATH "" CACHE PATH "Path to a prebuilt TensorFlow Lite for Microcontrollers library.") # Select accelerator for tensorflow @@ -58,8 +58,8 @@ if (CORE_SOFTWARE_ACCELERATOR STREQUAL "NPU") target_link_libraries(ethosu_core INTERFACE ethosu_core_driver) endif() -# Build Tensorflow library -include(tensorflow.cmake) +# Build Tensorflow Lite Micro library +include(tflite_micro.cmake) # Build RTOS add_subdirectory(rtos) -- cgit v1.2.1