Searched defs:gpu_allocator (Results 1 – 5 of 5) sorted by relevance
34 Allocator* gpu_allocator, Allocator* cpu_allocator) in GPUDevice()67 const string& physical_device_desc, Allocator* gpu_allocator, in CreateGPUDevice()
198 Allocator* gpu_allocator = gpu_bfc_allocator.get(); in GetGPUAllocator() local
391 Allocator* gpu_allocator, Allocator* cpu_allocator, in BaseGPUDevice()1592 Allocator* gpu_allocator = process_state->GetGPUAllocator( in CreateGPUDevice() local
305 Allocator* gpu_allocator = device->GetAllocator(value.alloc_attrs); in ComputeAsync() local
157 Allocator* gpu_allocator() { return gpu_allocator_; } in gpu_allocator() function in tensorflow::GPUDeviceTestHelper