Home
last modified time | relevance | path

Searched refs:name_to_input_name (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Dgraph_util_impl.py118 name_to_input_name = {} # Keyed by the dest node name.
128 name_to_input_name[n] = [_node_name(x) for x in node.input]
131 return name_to_input_name, name_to_node, name_to_seq_num
140 def _bfs_for_reachable_nodes(target_nodes, name_to_input_name): argument
152 if node in name_to_input_name:
153 next_to_visit += name_to_input_name[node]
180 name_to_input_name, name_to_node, name_to_seq_num = _extract_graph_summary(
184 nodes_to_keep = _bfs_for_reachable_nodes(dest_nodes, name_to_input_name)
/external/tensorflow/tensorflow/contrib/framework/python/framework/
Dgraph_util.py65 name_to_input_name, name_to_node, name_to_seq_num = _extract_graph_summary(
70 reachable_by_input = _bfs_for_reachable_nodes(input_nodes, name_to_input_name)
73 name_to_input_name)
98 input_node for input_node in name_to_input_name[cur_node]
/external/tensorflow/tensorflow/lite/python/
Dop_hint.py832 name_to_input_name, _, _ = _extract_graph_summary(graph_def)
835 reachable_by_input = _bfs_for_reachable_nodes(input_names, name_to_input_name)
837 name_to_input_name)
894 name_to_input_name): argument
917 input_node for input_node in name_to_input_name[current_node]
946 name_to_input_name, name_to_node, name_to_seq_num = _extract_graph_summary(
950 reachable_by_input = _bfs_for_reachable_nodes(input_names, name_to_input_name)
952 name_to_input_name)
1019 for input_name in name_to_input_name[n]:
1042 name_to_input_name, name_to_node, name_to_seq_num = _extract_graph_summary(
[all …]
Dconvert_test.py212 name_to_input_name, name_to_node, _ = (
215 used_node_names = _bfs_for_reachable_nodes(output_nodes, name_to_input_name)