Home
last modified time | relevance | path

Searched refs:required_nodes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dobfuscate_names.cc33 std::unordered_set<string> required_nodes; in ObfuscateNames() local
35 required_nodes.insert(input); in ObfuscateNames()
38 required_nodes.insert(output); in ObfuscateNames()
50 if (required_nodes.count(old_name)) { in ObfuscateNames()
66 } while (required_nodes.count(new_name)); in ObfuscateNames()
Dremove_nodes.cc43 std::set<string> required_nodes; in RemoveNodes() local
45 required_nodes.insert(NodeNameFromInput(input)); in RemoveNodes()
48 required_nodes.insert(NodeNameFromInput(output)); in RemoveNodes()
69 [&inputs_to_rename, &required_nodes, &any_nodes_removed]( in RemoveNodes()
76 if (required_nodes.count(replace_node.name())) { in RemoveNodes()
Dstrip_unused_nodes.cc119 std::set<string> required_nodes; in StripUnusedNodes() local
122 required_nodes.insert(NodeNameFromInput(input)); in StripUnusedNodes()
126 required_nodes.insert(output); in StripUnusedNodes()
140 required_nodes.insert(current_input); in StripUnusedNodes()
151 if (!required_nodes.count(input_node_name)) { in StripUnusedNodes()
163 return required_nodes.count(node.name()) > 0; in StripUnusedNodes()