fix arena

rkuester · Nov 4, 2024 · 40e7530 · 40e7530
1 parent 0d889e0
commit 40e7530
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tensorflow/lite/micro/tools/benchmarking/generic_model_benchmark.cc b/tensorflow/lite/micro/tools/benchmarking/generic_model_benchmark.cc
@@ -196,8 +196,7 @@ int Benchmark(const uint8_t* model_data, tflite::PrettyPrintType print_type) {
 
   event_handle = profiler.BeginEvent("tflite::RecordingMicroAllocator::Create");
   tflite::RecordingMicroAllocator* allocator(
-      tflite::RecordingMicroAllocator::Create(
-          reinterpret_cast<uint8_t*>(0xe1000000), kTensorArenaSize));
+      tflite::RecordingMicroAllocator::Create(tensor_arena, kTensorArenaSize));
   profiler.EndEvent(event_handle);
   event_handle = profiler.BeginEvent("tflite::MicroInterpreter instantiation");
   tflite::RecordingMicroInterpreter interpreter(