Searched refs:GPUcudaMallocAllocator (Results 1 – 3 of 3) sorted by relevance
29 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 absl::optional<AllocatorStats> GPUcudaMallocAllocator::GetStats() { in GetStats()68 bool GPUcudaMallocAllocator::TracksAllocationSizes() const { return false; } in TracksAllocationSizes()
32 class GPUcudaMallocAllocator : public Allocator {34 explicit GPUcudaMallocAllocator(Allocator* allocator,36 ~GPUcudaMallocAllocator() override;48 TF_DISALLOW_COPY_AND_ASSIGN(GPUcudaMallocAllocator);
191 new GPUcudaMallocAllocator(gpu_allocator, platform_gpu_id); in GetGPUAllocator()