Searched refs:to_display (Results 1 – 2 of 2) sorted by relevance
139 to_display = ['Layer (type)', 'Output Shape', 'Param #']146 to_display = ['Layer (type)', 'Output Shape', 'Param #', 'Connected to']163 print_row(to_display, positions)
1390 std::unordered_set<const HloInstruction*> to_display = {from, to}; in MakeNodeFromToFilter() local1391 while (!queue.empty() && to_display.size() < max_nodes) { in MakeNodeFromToFilter()1401 for (; it != path.end() && to_display.size() < max_nodes; ++it) { in MakeNodeFromToFilter()1402 to_display.insert(*it); in MakeNodeFromToFilter()1419 return to_display.count(instr) ? kNormalNode : kHideNode; in MakeNodeFromToFilter()