From 746dd103729735c842bc87c3b6323f7c3b7857ee Mon Sep 17 00:00:00 2001 From: Terry Heo Date: Thu, 26 Mar 2020 01:03:44 -0700 Subject: [PATCH] Use std::vector instead of std:list for ArenaAllocWithUsageInterval container When the number of tensors are is 1000, it affects about 80ms. PiperOrigin-RevId: 303060329 Change-Id: Ibd70d04f6a216f3b2e87c0ea3143b6e148d8049c --- tensorflow/lite/simple_memory_arena.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/lite/simple_memory_arena.h b/tensorflow/lite/simple_memory_arena.h index bb9407b7bfd..803b8be174f 100644 --- a/tensorflow/lite/simple_memory_arena.h +++ b/tensorflow/lite/simple_memory_arena.h @@ -16,8 +16,8 @@ limitations under the License. #define TENSORFLOW_LITE_SIMPLE_MEMORY_ARENA_H_ #include -#include #include +#include #include "tensorflow/lite/c/common.h" @@ -112,7 +112,7 @@ class SimpleMemoryArena { std::unique_ptr underlying_buffer_; size_t underlying_buffer_size_; char* underlying_buffer_aligned_ptr_; - std::list ordered_allocs_; + std::vector ordered_allocs_; }; } // namespace tflite