Searched defs:gpu_allocator (Results 1 – 7 of 7) sorted by relevance
34 Allocator* gpu_allocator, Allocator* cpu_allocator) in GPUDevice()67 const string& physical_device_desc, Allocator* gpu_allocator, in CreateGPUDevice()
117 Allocator* gpu_allocator = gpu_bfc_allocator; in GetGPUAllocator() local
318 Allocator* gpu_allocator, Allocator* cpu_allocator, in BaseGPUDevice()1220 Allocator* gpu_allocator = process_state->GetGPUAllocator( in CreateGPUDevice() local
302 Allocator* gpu_allocator() { return gpu_allocator_; } in gpu_allocator() function in tensorflow::GPUDeviceTestHelper
22 mod gpu_allocator; module
44 gpu_allocator: Option<Box<dyn GpuMemoryAllocator>>, field
304 Allocator* gpu_allocator = device->GetAllocator(value.alloc_attrs); in ComputeAsync() local