diff options
author | Davide Grohmann <davide.grohmann@arm.com> | 2022-08-08 17:12:08 +0200 |
---|---|---|
committer | Davide Grohmann <davide.grohmann@arm.com> | 2022-08-17 09:56:14 +0200 |
commit | 5508acb0c7803769b9bbda08762b232c13272bd9 (patch) | |
tree | cf62375a3d12e59c821318a13abaa510792fff69 /applications/baremetal/main.cpp | |
parent | 06f954bb6dd56921f1fb38e3e36405e3f1c70ef9 (diff) | |
download | ethos-u-core-platform-5508acb0c7803769b9bbda08762b232c13272bd9.tar.gz |
Generalize test models to be reused
Change-Id: I47705cfde2788fa74f8bbcd07100188e4690dbf2
Diffstat (limited to 'applications/baremetal/main.cpp')
-rw-r--r-- | applications/baremetal/main.cpp | 8 |
1 files changed, 2 insertions, 6 deletions
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 @@ using namespace InferenceProcess; * InferenceJob ****************************************************************************/ -#ifndef TENSOR_ARENA_SIZE -#define TENSOR_ARENA_SIZE 2000000 -#endif - -__attribute__((section(".bss.tensor_arena"), aligned(16))) uint8_t TFLuTensorArena[TENSOR_ARENA_SIZE]; +__attribute__((section(".bss.tensor_arena"), aligned(16))) uint8_t TFLuTensorArena[tensorArenaSize]; -class InferenceProcess inferenceProcess(TFLuTensorArena, TENSOR_ARENA_SIZE); +class InferenceProcess inferenceProcess(TFLuTensorArena, tensorArenaSize); uint8_t outputData[sizeof(expectedOutputData)] __attribute__((aligned(16), section("output_data_sec"))); |