Searched refs:GPUNanResetAllocator (Results 1 – 4 of 4) sorted by relevance
157 GPUNanResetAllocator::GPUNanResetAllocator(Allocator* allocator, in GPUNanResetAllocator() function in tensorflow::GPUNanResetAllocator165 GPUNanResetAllocator::~GPUNanResetAllocator() { delete base_allocator_; } in ~GPUNanResetAllocator()167 void* GPUNanResetAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw()186 void GPUNanResetAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw()205 size_t GPUNanResetAllocator::RequestedSize(const void* ptr) const { in RequestedSize()209 size_t GPUNanResetAllocator::AllocatedSize(const void* ptr) const { in AllocatedSize()213 absl::optional<AllocatorStats> GPUNanResetAllocator::GetStats() { in GetStats()217 void GPUNanResetAllocator::ClearStats() { base_allocator_->ClearStats(); } in ClearStats()
64 class GPUNanResetAllocator : public Allocator {66 explicit GPUNanResetAllocator(Allocator* allocator,68 ~GPUNanResetAllocator() override;82 TF_DISALLOW_COPY_AND_ASSIGN(GPUNanResetAllocator);
148 GPUNanResetAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()191 GPUNanResetAllocator a( in TEST()245 GPUNanResetAllocator a( in TEST()
184 gpu_allocator = new GPUNanResetAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()