Searched refs:GPUcudaMallocAllocator (Results 1 – 3 of 3) sorted by relevance
30 GPUcudaMallocAllocator::GPUcudaMallocAllocator(Allocator* allocator, in GPUcudaMallocAllocator() function in tensorflow::GPUcudaMallocAllocator37 GPUcudaMallocAllocator::~GPUcudaMallocAllocator() { delete base_allocator_; } in ~GPUcudaMallocAllocator()39 void* GPUcudaMallocAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw()54 void GPUcudaMallocAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw()64 bool GPUcudaMallocAllocator::TracksAllocationSizes() { return false; } in TracksAllocationSizes()
32 class GPUcudaMallocAllocator : public Allocator {34 explicit GPUcudaMallocAllocator(Allocator* allocator,36 ~GPUcudaMallocAllocator() override;47 TF_DISALLOW_COPY_AND_ASSIGN(GPUcudaMallocAllocator);
134 new GPUcudaMallocAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()