commit 5508acb0c7803769b9bbda08762b232c13272bd9
author Davide Grohmann <davide.grohmann@arm.com> Mon Aug 08 17:12:08 2022 +0200
committer Davide Grohmann <davide.grohmann@arm.com> Wed Aug 17 09:56:14 2022 +0200
tree cf62375a3d12e59c821318a13abaa510792fff69
parent 06f954bb6dd56921f1fb38e3e36405e3f1c70ef9

Generalize test models to be reused

Change-Id: I47705cfde2788fa74f8bbcd07100188e4690dbf2

applications/baremetal/main.cpp

diff --git a/applications/baremetal/main.cpp b/applications/baremetal/main.cpp
index e9b398a..ab365c6 100644
--- a/applications/baremetal/main.cpp
+++ b/applications/baremetal/main.cpp
@@ -45,13 +45,9 @@
  * InferenceJob
  ****************************************************************************/
 
-#ifndef TENSOR_ARENA_SIZE
-#define TENSOR_ARENA_SIZE 2000000
-#endif
+__attribute__((section(".bss.tensor_arena"), aligned(16))) uint8_t TFLuTensorArena[tensorArenaSize];
 
-__attribute__((section(".bss.tensor_arena"), aligned(16))) uint8_t TFLuTensorArena[TENSOR_ARENA_SIZE];
-
-class InferenceProcess inferenceProcess(TFLuTensorArena, TENSOR_ARENA_SIZE);
+class InferenceProcess inferenceProcess(TFLuTensorArena, tensorArenaSize);
 
 uint8_t outputData[sizeof(expectedOutputData)] __attribute__((aligned(16), section("output_data_sec")));