Searched refs:GPUNanResetAllocator (Results 1 – 4 of 4) sorted by relevance
156 GPUNanResetAllocator::GPUNanResetAllocator(Allocator* allocator, in GPUNanResetAllocator() function in tensorflow::GPUNanResetAllocator163 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()
64 class GPUNanResetAllocator : public Allocator {66 explicit GPUNanResetAllocator(Allocator* allocator,68 ~GPUNanResetAllocator() override;82 TF_DISALLOW_COPY_AND_ASSIGN(GPUNanResetAllocator);
138 GPUNanResetAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()183 GPUNanResetAllocator a( in TEST()239 GPUNanResetAllocator a( in TEST()
128 gpu_allocator = new GPUNanResetAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()