Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer.cc951 std::unordered_map<NodeDef*, SwapInfo>* nodes_to_swap) { in IdentifySwappingCandidates() argument
1101 (*nodes_to_swap)[fanout_to_swap.node].inputs_to_swap.push_back( in IdentifySwappingCandidates()
1117 std::unordered_map<NodeDef*, SwapInfo> nodes_to_swap; in SwappingPass() local
1122 IdentifySwappingCandidates(cluster, item, skip_list, &nodes_to_swap); in SwappingPass()
1127 SwapInfo& swap_info = nodes_to_swap[&node]; in SwappingPass()
1139 if (nodes_to_swap.empty()) { in SwappingPass()
1149 for (auto& swap : nodes_to_swap) { in SwappingPass()
1176 for (auto& swap : nodes_to_swap) { in SwappingPass()