Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_process_state.h107 virtual void AddGpuHostFreeVisitor(int numa_node,
Dgpu_process_state.cc305 void GPUProcessState::AddGpuHostFreeVisitor( in AddGpuHostFreeVisitor() function in tensorflow::GPUProcessState
/external/tensorflow/tensorflow/contrib/verbs/
Drdma_mgr.cc281 GPUProcessState::singleton()->AddGpuHostFreeVisitor(0, free_visitor); in RegMemVisitors()
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_memory_manager.cc255 GPUProcessState::singleton()->AddGpuHostFreeVisitor(numa_idx, free_visitor); in Init()