Searched refs:to_display (Results 1 – 2 of 2) sorted by relevance
169 to_display = ['Layer (type)', 'Output Shape', 'Param #']176 to_display = ['Layer (type)', 'Output Shape', 'Param #', 'Connected to']193 print_row(to_display, positions)
1429 std::unordered_set<const HloInstruction*> to_display = {from, to}; in MakeNodeFromToFilter() local1430 while (!queue.empty() && to_display.size() < max_nodes) { in MakeNodeFromToFilter()1440 for (; it != path.end() && to_display.size() < max_nodes; ++it) { in MakeNodeFromToFilter()1441 to_display.insert(*it); in MakeNodeFromToFilter()1458 return to_display.count(instr) ? kNormalNode : kHideNode; in MakeNodeFromToFilter()