Searched refs:FindCNode (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/allreduce_fusion/ |
| D | allreduce_fusion.h | 57 CNodeCostMap FindCNode(const AnfNodePtr &from, uint64_t recursive_times = 0) const;
|
| D | allreduce_fusion.cc | 86 CNodeCostMap AllreduceFusion::FindCNode(const AnfNodePtr &from, uint64_t recursive_times) const { in FindCNode() function in mindspore::parallel::AllreduceFusion
136 auto cnode_dist = FindCNode(input_node, recursive_times + 1); in FindNextCNodes()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.h | 87 std::pair<bool, CNodePtr> FindCNode(const AnfNodePtr &anode, const std::string &name, const FuncGra…
|
| D | step_parallel.cc | 994 std::pair<bool, CNodePtr> FindCNode(const AnfNodePtr &anode, const std::string &name, const FuncGra… in FindCNode() function
1028 return FindCNode(node_pair.first, name, func_graph, max_depth + 1); in FindCNode()
1174 auto next_cnode = FindCNode(param_node_pair.first, mirror_op_name, func_graph, 0); in InsertMirrorOps()
1589 auto next_cnode = FindCNode(parameter, op_name, cnode->func_graph(), 0); in ApplyParallelOptOnParam()
|