Searched defs:GetGPUAllocator (Results 1 – 2 of 2) sorted by relevance
88 Allocator* GetGPUAllocator(TfDeviceId tf_device_id) { in GetGPUAllocator() function
151 Allocator* GPUProcessState::GetGPUAllocator( in GetGPUAllocator() function in tensorflow::GPUProcessState