Searched refs:GPUProcessState (Results 1 – 11 of 11) sorted by relevance
59 /*static*/ GPUProcessState* GPUProcessState::singleton(GPUProcessState* ps) { in singleton()60 static GPUProcessState* instance = ps ? ps : new GPUProcessState; in singleton()66 GPUProcessState::GPUProcessState() : gpu_device_enabled_(false) { in GPUProcessState() function in tensorflow::GPUProcessState70 int GPUProcessState::BusIdForGPU(TfGpuId tf_gpu_id) { in BusIdForGPU()80 Allocator* GPUProcessState::GetGPUAllocator(const GPUOptions& options, in GetGPUAllocator()164 SharedCounter* GPUProcessState::GPUAllocatorCounter(TfGpuId tf_gpu_id) { in GPUAllocatorCounter()180 Allocator* GPUProcessState::GetGpuHostAllocator(int numa_node) { in GetGpuHostAllocator()276 void GPUProcessState::AddGPUAllocVisitor(int bus_id, in AddGPUAllocVisitor()291 void GPUProcessState::AddGpuHostAllocVisitor( in AddGpuHostAllocVisitor()305 void GPUProcessState::AddGpuHostFreeVisitor( in AddGpuHostFreeVisitor()[all …]
40 class GPUProcessState {54 static GPUProcessState* singleton(GPUProcessState* ps = nullptr);118 GPUProcessState();119 virtual ~GPUProcessState() {} in ~GPUProcessState()131 static GPUProcessState* instance_;
47 GPUProcessState* ps = GPUProcessState::singleton(); in GetAllocator()95 GPUProcessState* ps = GPUProcessState::singleton(); in GetAllocator()
313 GPUProcessState::singleton()->EnableGPUDevice(); in BaseGPUDevice()325 GPUProcessState::singleton()->GPUAllocatorCounter(tf_gpu_id); in BaseGPUDevice()1198 GPUProcessState* process_state = GPUProcessState::singleton(); in CreateGPUDevice()
65 void TearDown() override { GPUProcessState::singleton()->TestOnlyReset(); } in TearDown()
153 alloc = GPUProcessState::singleton()->GetGpuHostAllocator(0); in SetProtoFromGPU()
280 GPUProcessState::singleton()->AddGpuHostAllocVisitor(0, alloc_visitor); in RegMemVisitors()281 GPUProcessState::singleton()->AddGpuHostFreeVisitor(0, free_visitor); in RegMemVisitors()298 GPUProcessState::singleton()->AddGPUAllocVisitor(bus_id, in RegMemVisitors()
1089 Allocator* alloc = GPUProcessState::singleton()->GetGpuHostAllocator(0); in RecvHandler()1546 new Tensor(GPUProcessState::singleton()->GetGpuHostAllocator(0), in AllocateTensors()
82 friend class GPUProcessState; variable
253 GPUProcessState::singleton()->AddGpuHostAllocVisitor(numa_idx, in Init()255 GPUProcessState::singleton()->AddGpuHostFreeVisitor(numa_idx, free_visitor); in Init()264 GPUProcessState::singleton()->AddGPUAllocVisitor(numa_node_, in Init()
749 dev_allocator = GPUProcessState::singleton()->GetGPUAllocator( in GetDeviceAndAllocator()