Searched refs:repl_node (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | branch_culling.cc | 117 … std::unordered_map<AnfNodePtr, AnfNodePtr> repl_node, NodeInputReplMap repl_node_inputs, in RunSwitchNodeReplace() argument
126 if (repl_node.count(input) != 0) { in RunSwitchNodeReplace()
127 new_node->add_input(repl_node[input]); in RunSwitchNodeReplace()
136 for (auto &item : repl_node) { in RunSwitchNodeReplace()
156 std::unordered_map<AnfNodePtr, AnfNodePtr> repl_node; in TransformGraphCondBranchNodes() local
194 repl_node[node] = new_node; in TransformGraphCondBranchNodes()
198 RunSwitchNodeReplace(manager, nodes_changed, repl_node, repl_node_inputs, graph); in TransformGraphCondBranchNodes()
325 const std::shared_ptr<std::unordered_map<AnfNodePtr, AnfNodePtr>> &repl_node, in GenerateReplNodeForDependMakeTuple() argument
352 (*repl_node)[depended_node] = make_tuple_op; in GenerateReplNodeForDependMakeTuple()
359 const std::shared_ptr<std::unordered_map<AnfNodePtr, AnfNodePtr>> &repl_node, in GenerateRepDepend() argument
[all …]
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | func_graph_cloner.cc | 473 auto &repl_node = repl_map_node_[func_graph]; in SetEdges() local
474 if (repl_node.find(input) != repl_node.end()) { in SetEdges()
475 tx->SetEdge(cnode, SizeToInt(i), repl_node[input]); in SetEdges()
|