Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/graph_transforms/
Dtransform_utils.cc32 std::set<string>* matched_nodes) { in RecordMatchedNodes() argument
33 matched_nodes->insert(match.node.name()); in RecordMatchedNodes()
35 RecordMatchedNodes(input_match, matched_nodes); in RecordMatchedNodes()
291 std::set<string> matched_nodes; in GetOpTypeMatches() local
294 if (matched_nodes.count(node.name())) { in GetOpTypeMatches()
298 if (DoesOpTypeMatch(node, pattern, matched_nodes, &match)) { in GetOpTypeMatches()
299 RecordMatchedNodes(match, &matched_nodes); in GetOpTypeMatches()
374 std::set<string> matched_nodes; in ReplaceMatchingOpTypes() local
378 RecordMatchedNodes(match, &matched_nodes); in ReplaceMatchingOpTypes()
462 } else if (!matched_nodes.count(input_node.name())) { in ReplaceMatchingOpTypes()