Searched defs:gpu_allocator (Results 1 – 4 of 4) sorted by relevance
33 Allocator* gpu_allocator, Allocator* cpu_allocator) in GPUDevice()66 const string& physical_device_desc, Allocator* gpu_allocator, in CreateGPUDevice()
117 Allocator* gpu_allocator = gpu_bfc_allocator; in GetGPUAllocator() local
302 Allocator* gpu_allocator, Allocator* cpu_allocator, in BaseGPUDevice()1199 Allocator* gpu_allocator = process_state->GetGPUAllocator( in CreateGPUDevice() local
304 Allocator* gpu_allocator = device->GetAllocator(value.alloc_attrs); in ComputeAsync() local