Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer.cc112 std::unordered_set<const NodeDef*>* expanded_nodes) { in connected_subgraph() argument
114 for (const NodeDef* starting_node : *expanded_nodes) { in connected_subgraph()
117 expanded_nodes->clear(); in connected_subgraph()
121 if (!expanded_nodes->insert(current_node).second) { in connected_subgraph()
129 if (expanded_nodes->count(input_node) == 0 && in connected_subgraph()
137 if (expanded_nodes->count(output) == 0 && is_candidate(*output)) { in connected_subgraph()