Searched refs:sub_allocator_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_cpu_allocator.h | 57 : sub_allocator_(sub_allocator), name_(name) { in MklSmallSizeAllocator()
117 SubAllocator* sub_allocator_; // Not owned by this class. variable
181 sub_allocator_ = new MklSubAllocator(); in Initialize()
186 new MklSmallSizeAllocator(sub_allocator_, max_mem_bytes, kName); in Initialize()
188 new BFCAllocator(sub_allocator_, max_mem_bytes, kAllowGrowth, kName); in Initialize()
315 SubAllocator* sub_allocator_; // not owned by this class variable
|
| D | bfc_allocator.cc | 45 sub_allocator_(sub_allocator), in BFCAllocator()
86 sub_allocator_->Free(region.ptr(), region.memory_size()); in ~BFCAllocator()
129 void* mem_addr = sub_allocator_->Alloc(alignment, bytes, &bytes_received); in Extend()
140 mem_addr = sub_allocator_->Alloc(alignment, bytes, &bytes_received); in Extend()
382 sub_allocator_->Free(it->ptr(), it->memory_size()); in DeallocateRegions()
|
| D | bfc_allocator.h | 584 std::unique_ptr<SubAllocator> sub_allocator_; variable
|