Searched refs:AddCPUAllocVisitor (Results 1 – 4 of 4) sorted by relevance
72 void AddCPUAllocVisitor(SubAllocator::Visitor v);
125 void ProcessState::AddCPUAllocVisitor(SubAllocator::Visitor visitor) { in AddCPUAllocVisitor() function in tensorflow::ProcessState
276 ProcessState::singleton()->AddCPUAllocVisitor(alloc_visitor); in RegMemVisitors()
248 ProcessState::singleton()->AddCPUAllocVisitor(alloc_visitor); in Init()