Searched refs:GetGpuHostAllocator (Results 1 – 5 of 5) sorted by relevance
48 return ps->GetGpuHostAllocator(0); in GetAllocator()97 return ps->GetGpuHostAllocator(numa_node_); in GetAllocator()
86 virtual Allocator* GetGpuHostAllocator(int numa_node);
180 Allocator* GPUProcessState::GetGpuHostAllocator(int numa_node) { in GetGpuHostAllocator() function in tensorflow::GPUProcessState
153 alloc = GPUProcessState::singleton()->GetGpuHostAllocator(0); in SetProtoFromGPU()
1089 Allocator* alloc = GPUProcessState::singleton()->GetGpuHostAllocator(0); in RecvHandler()1546 new Tensor(GPUProcessState::singleton()->GetGpuHostAllocator(0), in AllocateTensors()