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()
313 SubAllocator* sub_allocator_; // not owned by this class variable
|
| D | bfc_allocator.cc | 35 : sub_allocator_(sub_allocator), in BFCAllocator()
76 sub_allocator_->Free(region.ptr(), region.memory_size()); in ~BFCAllocator()
112 void* mem_addr = sub_allocator_->Alloc(alignment, bytes); in Extend()
123 mem_addr = sub_allocator_->Alloc(alignment, bytes); in Extend()
|
| D | bfc_allocator.h | 438 std::unique_ptr<SubAllocator> sub_allocator_; variable
|