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
86 GPUDebugAllocator::~GPUDebugAllocator() { delete base_allocator_; } in ~GPUDebugAllocator()
88 void* GPUDebugAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw()
106 void GPUDebugAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw()
118 bool GPUDebugAllocator::TracksAllocationSizes() const { return true; } in TracksAllocationSizes()
120 size_t GPUDebugAllocator::RequestedSize(const void* ptr) const { in RequestedSize()
126 size_t GPUDebugAllocator::AllocatedSize(const void* ptr) const { in AllocatedSize()
131 int64 GPUDebugAllocator::AllocationId(const void* ptr) const { in AllocationId()
136 absl::optional<AllocatorStats> GPUDebugAllocator::GetStats() { in GetStats()
140 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.cc52 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
80 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
116 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
192 new GPUDebugAllocator(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
235 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
246 new GPUDebugAllocator(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
Dgpu_process_state.cc183 gpu_allocator = new GPUDebugAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()