diff options
author | 2018-01-13 16:21:55 -0800 | |
---|---|---|
committer | 2018-01-13 16:25:51 -0800 | |
commit | e6ff665dbe4888aa5fdff8f34c44405acca2ddd1 (patch) | |
tree | a290fe97b76bebec82ef3b3a76c906acf6b55f41 /tensorflow/contrib/lite/simple_memory_arena.h | |
parent | a4973345351a14a786987cd7f648a99c029fdc1d (diff) |
Clean up the allocation logic in the interpreter.
PiperOrigin-RevId: 181865795
Diffstat (limited to 'tensorflow/contrib/lite/simple_memory_arena.h')
-rw-r--r-- | tensorflow/contrib/lite/simple_memory_arena.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/simple_memory_arena.h b/tensorflow/contrib/lite/simple_memory_arena.h index 0d0b7f9ff7..07a38c4243 100644 --- a/tensorflow/contrib/lite/simple_memory_arena.h +++ b/tensorflow/contrib/lite/simple_memory_arena.h @@ -68,6 +68,10 @@ class SimpleMemoryArena { TfLiteStatus Clear(); + int64_t BasePointer() const { + return reinterpret_cast<int64_t>(underlying_buffer_aligned_ptr_); + } + private: bool commited_; size_t arena_alignment_; |