Searched refs:AddGpuHostFreeVisitor (Results 1 – 4 of 4) sorted by relevance
107 virtual void AddGpuHostFreeVisitor(int numa_node,
305 void GPUProcessState::AddGpuHostFreeVisitor( in AddGpuHostFreeVisitor() function in tensorflow::GPUProcessState
281 GPUProcessState::singleton()->AddGpuHostFreeVisitor(0, free_visitor); in RegMemVisitors()
255 GPUProcessState::singleton()->AddGpuHostFreeVisitor(numa_idx, free_visitor); in Init()