Searched refs:all_node_names (Results 1 – 2 of 2) sorted by relevance
34 self.all_node_names = []38 self.all_node_names.append(pytree_utils.NodeName(node))42 self.all_node_names.append(pytree_utils.NodeName(leaf))73 self.assertEqual(expected_names, collector.all_node_names)90 self.assertEqual(expected_names, collector.all_node_names)
311 absl::flat_hash_map<string, int> all_node_names; in FindSinkNode() local314 all_node_names.insert_or_assign(graph_def.node(i).name(), i); in FindSinkNode()326 const NodeDef& sink_graph_node = graph_def.node(all_node_names[it.first]); in FindSinkNode()