Searched refs:GetGpuHostAllocator (Results 1 – 6 of 6) sorted by relevance
49 return ps->GetGpuHostAllocator(0); in GetAllocator()98 return ps->GetGpuHostAllocator(numa_node_); in GetAllocator()
93 virtual Allocator* GetGpuHostAllocator(int numa_node);
246 Allocator* GPUProcessState::GetGpuHostAllocator(int numa_node) { in GetGpuHostAllocator() function in tensorflow::GPUProcessState
151 alloc = GPUProcessState::singleton()->GetGpuHostAllocator(0); in SetProtoFromGPU()
161 std::unique_ptr<tensorflow::BFCAllocator> GetGpuHostAllocator( in GetGpuHostAllocator() function324 GetGpuHostAllocator(local_device_states.front()->executor()); in GetGpuClient()
153 host_allocator_ = GPUProcessState::singleton()->GetGpuHostAllocator(0); in GPUDeviceTestHelper()