Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_debug_allocator.cc156 GPUNanResetAllocator::GPUNanResetAllocator(Allocator* allocator, in GPUNanResetAllocator() function in tensorflow::GPUNanResetAllocator
163 GPUNanResetAllocator::~GPUNanResetAllocator() { delete base_allocator_; } in ~GPUNanResetAllocator()
165 void* GPUNanResetAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw()
184 void GPUNanResetAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw()
203 size_t GPUNanResetAllocator::RequestedSize(const void* ptr) { in RequestedSize()
207 size_t GPUNanResetAllocator::AllocatedSize(const void* ptr) { in AllocatedSize()
211 absl::optional<AllocatorStats> GPUNanResetAllocator::GetStats() { in GetStats()
215 void GPUNanResetAllocator::ClearStats() { base_allocator_->ClearStats(); } in ClearStats()
Dgpu_debug_allocator.h64 class GPUNanResetAllocator : public Allocator {
66 explicit GPUNanResetAllocator(Allocator* allocator,
68 ~GPUNanResetAllocator() override;
82 TF_DISALLOW_COPY_AND_ASSIGN(GPUNanResetAllocator);
Dgpu_debug_allocator_test.cc138 GPUNanResetAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
183 GPUNanResetAllocator a( in TEST()
239 GPUNanResetAllocator a( in TEST()
Dgpu_process_state.cc128 gpu_allocator = new GPUNanResetAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()