Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_process_state.h97 virtual void AddGPUAllocVisitor(int bus_id,
Dgpu_process_state.cc276 void GPUProcessState::AddGPUAllocVisitor(int bus_id, in AddGPUAllocVisitor() function in tensorflow::GPUProcessState
/external/tensorflow/tensorflow/contrib/verbs/
Drdma_mgr.cc298 GPUProcessState::singleton()->AddGPUAllocVisitor(bus_id, in RegMemVisitors()
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_memory_manager.cc264 GPUProcessState::singleton()->AddGPUAllocVisitor(numa_node_, in Init()