Home
last modified time | relevance | path

Searched refs:GPUDebugAllocator (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_debug_allocator.cc78 GPUDebugAllocator::GPUDebugAllocator(Allocator* allocator, in GPUDebugAllocator() function in tensorflow::GPUDebugAllocator
85 GPUDebugAllocator::~GPUDebugAllocator() { delete base_allocator_; } in ~GPUDebugAllocator()
87 void* GPUDebugAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw()
105 void GPUDebugAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw()
117 bool GPUDebugAllocator::TracksAllocationSizes() { return true; } in TracksAllocationSizes()
119 size_t GPUDebugAllocator::RequestedSize(const void* ptr) { in RequestedSize()
125 size_t GPUDebugAllocator::AllocatedSize(const void* ptr) { in AllocatedSize()
130 int64 GPUDebugAllocator::AllocationId(const void* ptr) { in AllocationId()
135 absl::optional<AllocatorStats> GPUDebugAllocator::GetStats() { in GetStats()
139 void GPUDebugAllocator::ClearStats() { base_allocator_->ClearStats(); } in ClearStats()
[all …]
Dgpu_debug_allocator.h34 class GPUDebugAllocator : public Allocator {
36 explicit GPUDebugAllocator(Allocator* allocator,
38 ~GPUDebugAllocator() override;
58 TF_DISALLOW_COPY_AND_ASSIGN(GPUDebugAllocator);
Dgpu_debug_allocator_test.cc41 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
69 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
105 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
184 new GPUDebugAllocator(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
229 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
240 new GPUDebugAllocator(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
Dgpu_process_state.cc127 gpu_allocator = new GPUDebugAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()