experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use std::ptrdiff_t instead of size_t for representing difference of pointers.

Browse Source 
Use size_t instead of int for representing size in memory.

PiperOrigin-RevId: 288378068
Change-Id: Id4490f1203908cb2650c3817e087f0d5bc3ee4fb
This commit is contained in:
Robert David 2020-01-06 14:35:55 -08:00 committed by TensorFlower Gardener
parent a87225391d
commit f9ef1a5844
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/lite/micro/simple_memory_allocator.cc
View File

@ -15,6 +15,8 @@ limitations under the License.
#include "tensorflow/lite/micro/simple_memory_allocator.h"
#include <cstddef>
#include "tensorflow/lite/core/api/flatbuffer_conversions.h"
#include "tensorflow/lite/micro/memory_helpers.h"
@ -29,7 +31,7 @@ uint8_t* SimpleMemoryAllocator::AllocateFromTail(size_t size,
uint8_t* previous_free = (data_ + data_size_max_) - data_size_;
uint8_t* current_data = previous_free - size;
uint8_t* aligned_result = AlignPointerDown(current_data, alignment);
size_t aligned_size = (previous_free - aligned_result);
std::ptrdiff_t aligned_size = (previous_free - aligned_result);
if ((data_size_ + aligned_size) > data_size_max_) {
// TODO(petewarden): Add error reporting beyond returning null!
return nullptr;

4
tensorflow/lite/micro/simple_memory_allocator.h
View File

@ -35,7 +35,7 @@ class SimpleMemoryAllocator {
// in ascending order.
uint8_t* AllocateFromTail(size_t size, size_t alignment);
int GetDataSize() const { return data_size_; }
size_t GetDataSize() const { return data_size_; }
// Child allocator is something like a temporary allocator. Memory allocated
// by the child allocator will be freed once the child allocator is
@ -50,7 +50,7 @@ class SimpleMemoryAllocator {
~SimpleMemoryAllocator();
private:
int data_size_ = 0;
size_t data_size_ = 0;
size_t data_size_max_;
uint8_t* data_;
SimpleMemoryAllocator* parent_allocator_ = nullptr;