Searched refs:GPUProcessState (Results 1 – 9 of 9) sorted by relevance
60 /*static*/ GPUProcessState* GPUProcessState::singleton(GPUProcessState* ps) { in singleton()61 static GPUProcessState* instance = ps ? ps : new GPUProcessState; in singleton()67 GPUProcessState::GPUProcessState() : gpu_device_enabled_(false) { in GPUProcessState() function in tensorflow::GPUProcessState71 int GPUProcessState::BusIdForGPU(TfGpuId tf_gpu_id) { in BusIdForGPU()136 Allocator* GPUProcessState::GetGPUAllocator( in GetGPUAllocator()221 SharedCounter* GPUProcessState::GPUAllocatorCounter(TfGpuId tf_gpu_id) { in GPUAllocatorCounter()246 Allocator* GPUProcessState::GetGpuHostAllocator(int numa_node) { in GetGpuHostAllocator()345 void GPUProcessState::AddGPUAllocVisitor(int bus_id, in AddGPUAllocVisitor()361 void GPUProcessState::AddGpuHostAllocVisitor( in AddGpuHostAllocVisitor()376 void GPUProcessState::AddGpuHostFreeVisitor( in AddGpuHostFreeVisitor()[all …]
41 class GPUProcessState {55 static GPUProcessState* singleton(GPUProcessState* ps = nullptr);125 GPUProcessState();126 virtual ~GPUProcessState() {} in ~GPUProcessState()138 static GPUProcessState* instance_;
48 GPUProcessState* ps = GPUProcessState::singleton(); in GetAllocator()96 GPUProcessState* ps = GPUProcessState::singleton(); in GetAllocator()
393 GPUProcessState::singleton()->EnableGPUDevice(); in BaseGPUDevice()464 GPUProcessState::singleton()->GPUAllocatorCounter(tf_gpu_id_); in Init()1431 GPUProcessState* process_state = GPUProcessState::singleton(); in CreateGPUDevice()
151 alloc = GPUProcessState::singleton()->GetGpuHostAllocator(0); in SetProtoFromGPU()
66 GPUProcessState::singleton()->TestOnlyReset(); in TearDown()
83 friend class GPUProcessState; variable
151 gpu_allocator_ = GPUProcessState::singleton()->GetGPUAllocator( in GPUDeviceTestHelper()153 host_allocator_ = GPUProcessState::singleton()->GetGpuHostAllocator(0); in GPUDeviceTestHelper()
647 dev_allocator = GPUProcessState::singleton()->GetGPUAllocator( in GetDeviceAndAllocator()