Searched refs:AddGpuHostAllocVisitor (Results 1 – 4 of 4) sorted by relevance
102 virtual void AddGpuHostAllocVisitor(int numa_node,
291 void GPUProcessState::AddGpuHostAllocVisitor( in AddGpuHostAllocVisitor() function in tensorflow::GPUProcessState
280 GPUProcessState::singleton()->AddGpuHostAllocVisitor(0, alloc_visitor); in RegMemVisitors()
253 GPUProcessState::singleton()->AddGpuHostAllocVisitor(numa_idx, in Init()