Searched refs:PoolAllocator (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | pool_allocator.cc | 36 PoolAllocator::PoolAllocator(size_t pool_size_limit, bool auto_resize, in PoolAllocator() function in tensorflow::PoolAllocator
51 PoolAllocator::~PoolAllocator() { Clear(); } in ~PoolAllocator()
95 void* PoolAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw()
135 void PoolAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw()
155 void PoolAllocator::Clear() { in Clear()
173 void PoolAllocator::RemoveFromList(PtrRecord* pr) { in RemoveFromList()
191 void PoolAllocator::AddToList(PtrRecord* pr) { in AddToList()
204 void PoolAllocator::EvictOne() { in EvictOne()
|
| D | pool_allocator.h | 43 class PoolAllocator : public Allocator {
55 PoolAllocator(size_t pool_size_limit, bool auto_resize,
58 ~PoolAllocator() override;
|
| D | process_state.h | 34 class PoolAllocator; variable
|
| D | process_state.cc | 103 new PoolAllocator(100 /*pool_size_limit*/, true /*auto_resize*/, in GetCPUAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | pool_allocator_test.cc | 30 PoolAllocator pool( in TEST()
49 PoolAllocator pool( in TEST()
83 PoolAllocator pool( in TEST()
102 PoolAllocator pool(2 /*pool_size_limit*/, true /*auto_resize*/, in TEST()
149 PoolAllocator pool(2 /*pool_size_limit*/, false /*auto_resize*/, in TEST()
248 PoolAllocator pool( in TEST()
|
| D | gpu_process_state.h | 36 class PoolAllocator; variable
|