Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer.cc1213 std::pair<NodeDef*, NodeDef*> swap_nodes; in SwappingPass() local
1214 if (!BuildSwapPair(node, input_id, name_map, &item->graph, &swap_nodes) in SwappingPass()
1218 *swap_nodes.first->add_input() = node->input(input_id); in SwappingPass()
1219 *node->mutable_input(input_id) = swap_nodes.second->name(); in SwappingPass()
1222 out_trigger->add_input(strings::StrCat("^", swap_nodes.first->name())); in SwappingPass()
1223 swap_nodes.second->add_input(strings::StrCat("^", in_trigger->name())); in SwappingPass()
1226 skip_list->insert(swap_nodes.first->name()); in SwappingPass()
1227 skip_list->insert(swap_nodes.second->name()); in SwappingPass()