diff options
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"))); |