Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dbfc_allocator.cc37 constexpr BFCAllocator::ChunkHandle BFCAllocator::kInvalidChunkHandle;
38 constexpr uint64 BFCAllocator::kMemDebugHistorySize;
40 BFCAllocator::BFCAllocator(SubAllocator* sub_allocator, size_t total_memory, in BFCAllocator() function in tensorflow::BFCAllocator
81 BFCAllocator::~BFCAllocator() { in ~BFCAllocator()
94 BFCAllocator::Chunk* BFCAllocator::ChunkFromHandle(ChunkHandle h) { in ChunkFromHandle()
100 const BFCAllocator::Chunk* BFCAllocator::ChunkFromHandle(ChunkHandle h) const { in ChunkFromHandle()
106 bool BFCAllocator::Extend(size_t alignment, size_t rounded_bytes) { in Extend()
174 BFCAllocator::Chunk* c = ChunkFromHandle(h); in Extend()
189 BFCAllocator::Chunk* prev_chunk = ChunkFromHandle(prev); in Extend()
205 BFCAllocator::ChunkHandle BFCAllocator::AllocateChunk() { in AllocateChunk()
[all …]
Dbfc_allocator.h49 class BFCAllocator : public Allocator {
52 BFCAllocator(SubAllocator* sub_allocator, size_t total_memory,
55 ~BFCAllocator() override;
210 string DebugString(BFCAllocator* a, in DebugString()
217 if (recurse && prev != BFCAllocator::kInvalidChunkHandle) { in DebugString()
221 if (recurse && next != BFCAllocator::kInvalidChunkHandle) { in DebugString()
240 explicit ChunkComparator(BFCAllocator* allocator) in ChunkComparator()
254 BFCAllocator* allocator_; // The parent allocator
261 Bin(BFCAllocator* allocator, size_t bs) in Bin()
614 TF_DISALLOW_COPY_AND_ASSIGN(BFCAllocator);
Dbfc_allocator_test.cc54 BFCAllocator bfc_allocator(sub_allocator, 1 << 30, false, "GPU_0_bfc"); in BM_Allocator()
Dprocess_state.cc104 new BFCAllocator(sub_allocator, cpu_mem_limit, /*allow_growth=*/true, 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.cc456 std::array<BFCAllocator::BinDebugInfo, BFCAllocator::kNumBins> bin_infos; in TestBinDebugInfo()
465 for (int i = 0; i < BFCAllocator::kNumBins; i++) { in TestBinDebugInfo()
466 const BFCAllocator::BinDebugInfo& bin_info = bin_infos[i]; in TestBinDebugInfo()
490 if (i == BFCAllocator::kNumBins - 1) { in TestBinDebugInfo()
513 for (int i = 0; i < BFCAllocator::kNumBins; i++) { in TestBinDebugInfo()
514 const BFCAllocator::BinDebugInfo& bin_info = bin_infos[i]; in TestBinDebugInfo()
697 for (int i = 0; i < BFCAllocator::kNumBins; i++) { in TestRegionDeallocation()
698 BFCAllocator::Bin* bin = a.BinFromIndex(i); in TestRegionDeallocation()
Dgpu_bfc_allocator.h35 class GPUBFCAllocator : public BFCAllocator {
Dgpu_bfc_allocator.cc84 : BFCAllocator(sub_allocator, total_memory, in GPUBFCAllocator()
Dgpu_process_state.cc313 new BFCAllocator(sub_allocator, gpu_host_mem_limit, in GetGpuHostAllocator()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_device.cc133 auto gpu_bfc_allocator = absl::make_unique<tensorflow::BFCAllocator>( in CreateBFCAllocator()
161 std::unique_ptr<tensorflow::BFCAllocator> GetGpuHostAllocator( in GetGpuHostAllocator()
167 return absl::make_unique<tensorflow::BFCAllocator>( in GetGpuHostAllocator()
/external/tensorflow/
DRELEASE.md3476 * Reduce BFCAllocator internal fragmentation.