Home
last modified time | relevance | path

Searched refs:GetGpuHostAllocator (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_factory.cc48 return ps->GetGpuHostAllocator(0); in GetAllocator()
97 return ps->GetGpuHostAllocator(numa_node_); in GetAllocator()
Dgpu_process_state.h86 virtual Allocator* GetGpuHostAllocator(int numa_node);
Dgpu_process_state.cc180 Allocator* GPUProcessState::GetGpuHostAllocator(int numa_node) { in GetGpuHostAllocator() function in tensorflow::GPUProcessState
Dgpu_util.cc153 alloc = GPUProcessState::singleton()->GetGpuHostAllocator(0); in SetProtoFromGPU()
/external/tensorflow/tensorflow/contrib/verbs/
Drdma.cc1089 Allocator* alloc = GPUProcessState::singleton()->GetGpuHostAllocator(0); in RecvHandler()
1546 new Tensor(GPUProcessState::singleton()->GetGpuHostAllocator(0), in AllocateTensors()