HomeSort by relevance Sort by last modified time
    Searched full:base_allocator_ (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_debug_allocator.cc 80 : base_allocator_(allocator) {
84 GPUDebugAllocator::~GPUDebugAllocator() { delete base_allocator_; }
89 void* allocated_ptr = base_allocator_->AllocateRaw(alignment, num_bytes);
98 size_t req_size = base_allocator_->RequestedSize(allocated_ptr);
111 base_allocator_->DeallocateRaw(ptr);
115 return base_allocator_->AddAllocVisitor(visitor);
119 return base_allocator_->AddFreeVisitor(visitor);
125 auto req_size = base_allocator_->RequestedSize(static_cast<const char*>(ptr) -
131 return base_allocator_->AllocatedSize(static_cast<const char*>(ptr) -
136 return base_allocator_->AllocationId(static_cast<const char*>(ptr)
    [all...]
gpu_cudamalloc_allocator.cc 32 : base_allocator_(allocator) {
36 GPUcudaMallocAllocator::~GPUcudaMallocAllocator() { delete base_allocator_; }
64 return base_allocator_->AddAllocVisitor(visitor);
68 return base_allocator_->AddFreeVisitor(visitor);
gpu_debug_allocator.h 56 VisitableAllocator* base_allocator_ = nullptr; // owned member in class:tensorflow::GPUDebugAllocator
82 VisitableAllocator* base_allocator_ = nullptr; // owned member in class:tensorflow::GPUNanResetAllocator
gpu_cudamalloc_allocator.h 45 VisitableAllocator* base_allocator_ = nullptr; // owned member in class:tensorflow::GPUcudaMallocAllocator

Completed in 147 milliseconds