Home
last modified time | relevance | path

Searched refs:AddGpuHostAllocVisitor (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_process_state.h102 virtual void AddGpuHostAllocVisitor(int numa_node,
Dgpu_process_state.cc291 void GPUProcessState::AddGpuHostAllocVisitor( in AddGpuHostAllocVisitor() function in tensorflow::GPUProcessState
/external/tensorflow/tensorflow/contrib/verbs/
Drdma_mgr.cc280 GPUProcessState::singleton()->AddGpuHostAllocVisitor(0, alloc_visitor); in RegMemVisitors()
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_memory_manager.cc253 GPUProcessState::singleton()->AddGpuHostAllocVisitor(numa_idx, in Init()