Home
last modified time | relevance | path

Searched refs:BFCAllocator (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dbfc_allocator.cc33 BFCAllocator::BFCAllocator(SubAllocator* sub_allocator, size_t total_memory, in BFCAllocator() function in tensorflow::BFCAllocator
71 BFCAllocator::~BFCAllocator() { in ~BFCAllocator()
84 BFCAllocator::Chunk* BFCAllocator::ChunkFromHandle(ChunkHandle h) { in ChunkFromHandle()
90 bool BFCAllocator::Extend(size_t alignment, size_t rounded_bytes) { in Extend()
150 BFCAllocator::Chunk* c = ChunkFromHandle(h); in Extend()
166 BFCAllocator::ChunkHandle BFCAllocator::AllocateChunk() { in AllocateChunk()
179 void BFCAllocator::DeallocateChunk(ChunkHandle h) { in DeallocateChunk()
185 void* BFCAllocator::AllocateRawInternalWithRetry( in AllocateRawInternalWithRetry()
212 void* BFCAllocator::AllocateRaw(size_t unused_alignment, size_t num_bytes, in AllocateRaw()
246 size_t BFCAllocator::RoundedBytes(size_t bytes) { in RoundedBytes()
[all …]
Dbfc_allocator.h46 class BFCAllocator : public Allocator {
49 BFCAllocator(SubAllocator* sub_allocator, size_t total_memory,
51 ~BFCAllocator() override;
148 string DebugString(BFCAllocator* a, in DebugString()
155 if (recurse && prev != BFCAllocator::kInvalidChunkHandle) { in DebugString()
159 if (recurse && next != BFCAllocator::kInvalidChunkHandle) { in DebugString()
173 explicit ChunkComparator(BFCAllocator* allocator) in ChunkComparator()
187 BFCAllocator* allocator_; // The parent allocator
194 Bin(BFCAllocator* allocator, size_t bs) in Bin()
459 TF_DISALLOW_COPY_AND_ASSIGN(BFCAllocator);
Dprocess_state.cc96 new BFCAllocator(sub_allocator, cpu_mem_limit, true /*allow_growth*/, in GetCPUAllocator()
Dmkl_cpu_allocator.h188 new BFCAllocator(sub_allocator_, max_mem_bytes, kAllowGrowth, kName); in Initialize()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_bfc_allocator_test.cc442 std::array<BFCAllocator::BinDebugInfo, BFCAllocator::kNumBins> bin_infos; in TestBinDebugInfo()
448 for (int i = 0; i < BFCAllocator::kNumBins; i++) { in TestBinDebugInfo()
449 const BFCAllocator::BinDebugInfo& bin_info = bin_infos[i]; in TestBinDebugInfo()
465 if (i == BFCAllocator::kNumBins - 1) { in TestBinDebugInfo()
483 for (int i = 0; i < BFCAllocator::kNumBins; i++) { in TestBinDebugInfo()
484 const BFCAllocator::BinDebugInfo& bin_info = bin_infos[i]; in TestBinDebugInfo()
Dgpu_bfc_allocator.cc66 : BFCAllocator(sub_allocator, total_memory, in GPUBFCAllocator()
Dgpu_bfc_allocator.h87 class GPUBFCAllocator : public BFCAllocator {
Dgpu_process_state.cc244 new BFCAllocator(sub_allocator, gpu_host_mem_limit, in GetGpuHostAllocator()
/external/tensorflow/
DRELEASE.md803 * Reduce BFCAllocator internal fragmentation.