Searched refs:GPUcudaMallocAllocator (Results 1 – 3 of 3) sorted by relevance
29 GPUcudaMallocAllocator::GPUcudaMallocAllocator( in GPUcudaMallocAllocator() function in tensorflow::GPUcudaMallocAllocator36 void* GPUcudaMallocAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw()59 void GPUcudaMallocAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw()83 bool GPUcudaMallocAllocator::TracksAllocationSizes() const { return false; } in TracksAllocationSizes()
31 class GPUcudaMallocAllocator : public Allocator {33 explicit GPUcudaMallocAllocator(PlatformDeviceId platform_device_id);46 TF_DISALLOW_COPY_AND_ASSIGN(GPUcudaMallocAllocator);
219 gpu_allocator = new GPUcudaMallocAllocator(platform_device_id); in GetGPUAllocator()