Searched refs:GetGPUAllocator (Results 1 – 4 of 4) sorted by relevance
83 virtual Allocator* GetGPUAllocator(const GPUOptions& options,
80 Allocator* GPUProcessState::GetGPUAllocator(const GPUOptions& options, in GetGPUAllocator() function in tensorflow::GPUProcessState
1199 Allocator* gpu_allocator = process_state->GetGPUAllocator( in CreateGPUDevice()
749 dev_allocator = GPUProcessState::singleton()->GetGPUAllocator( in GetDeviceAndAllocator()