Searched refs:name_to_input_name (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | graph_util_impl.py | 180 name_to_input_name = {} # Keyed by the dest node name.
190 name_to_input_name[n] = [_node_name(x) for x in node.input]
194 name_to_input_name[n].append(
198 return name_to_input_name, name_to_node, name_to_seq_num
207 def _bfs_for_reachable_nodes(target_nodes, name_to_input_name): argument
219 if node in name_to_input_name:
220 next_to_visit += name_to_input_name[node]
249 name_to_input_name, name_to_node, name_to_seq_num = _extract_graph_summary(
253 nodes_to_keep = _bfs_for_reachable_nodes(dest_nodes, name_to_input_name)
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | op_hint.py | 843 name_to_input_name, _, _ = _extract_graph_summary(graph_def)
846 reachable_by_input = _bfs_for_reachable_nodes(input_names, name_to_input_name)
848 name_to_input_name)
904 name_to_input_name): argument
927 input_node for input_node in name_to_input_name[current_node]
955 name_to_input_name, name_to_node, name_to_seq_num = _extract_graph_summary(
959 reachable_by_input = _bfs_for_reachable_nodes(input_names, name_to_input_name)
961 name_to_input_name)
1046 for input_name in name_to_input_name[n]:
1069 name_to_input_name, name_to_node, name_to_seq_num = _extract_graph_summary(
[all …]
|
| D | convert_test.py | 188 name_to_input_name, name_to_node, _ = (
191 used_node_names = _bfs_for_reachable_nodes(output_nodes, name_to_input_name)
|