From 66ed182cd7520537e73ec37f17f8bf549d8297a2 Mon Sep 17 00:00:00 2001 From: Anton Moberg Date: Wed, 10 Feb 2021 08:49:28 +0100 Subject: core_software - Move TensorArena Moved TensorArena in inference_process.ccp to application level. InferenceProcess class now takes TensorArena pointer and TensorArenaSize as parameters. Needs to be set by application before runJob() is called. Change-Id: I530b96039868305fa903ae7f93419d9d00f9c16f --- applications/inference_process/CMakeLists.txt | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) (limited to 'applications/inference_process/CMakeLists.txt') diff --git a/applications/inference_process/CMakeLists.txt b/applications/inference_process/CMakeLists.txt index 3348d28..97d4d8f 100644 --- a/applications/inference_process/CMakeLists.txt +++ b/applications/inference_process/CMakeLists.txt @@ -16,14 +16,11 @@ # limitations under the License. # -set(TR_ARENA_SIZE "200000" CACHE STRING "Arena size.") set(TR_PRINT_OUTPUT_BYTES "" CACHE STRING "Print output data.") add_library(inference_process STATIC) + target_include_directories(inference_process PUBLIC include PRIVATE ${TENSORFLOW_PATH} ${TENSORFLOW_PATH}/tensorflow/lite/micro/tools/make/downloads/flatbuffers/include) target_link_libraries(inference_process PUBLIC tflu cmsis_core cmsis_device) -target_sources(inference_process PRIVATE src/inference_process.cpp) - -# Set arena size -target_compile_definitions(inference_process PRIVATE TENSOR_ARENA_SIZE=${TR_ARENA_SIZE}) +target_sources(inference_process PRIVATE src/inference_process.cpp) \ No newline at end of file -- cgit v1.2.1