Searched refs:prior_node (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | pipeline_split_utils.cc | 276 void InsertDepend(const AnfNodePtr &prior_node, const AnfNodePtr &post_node, const FuncGraphManager… in InsertDepend() argument
278 MS_EXCEPTION_IF_NULL(prior_node); in InsertDepend()
282 …tor<AnfNodePtr> depend_input = {NewValueNode(prim::kPrimDepend), post_cnode->input(1), prior_node}; in InsertDepend()
296 auto prior_node = forward_end[i - 1]; in ReorderForForward() local
298 InsertDepend(prior_node, post_node, manager, root); in ReorderForForward()
349 auto prior_node = forward_params_pair.second[0]; in ReorderForParams() local
351 InsertDepend(prior_node, post_node, manager, root); in ReorderForParams()
398 auto prior_node = temp_vec[j]; in Deduplicate() local
400 InsertDepend(prior_node, post_node, manager, root); in Deduplicate()
|
| D | pipeline_split_utils.h | 48 void InsertDepend(const AnfNodePtr &prior_node, const AnfNodePtr &post_node, const FuncGraphManager…
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | parallel_fusion.cc | 401 auto prior_node = get_info(node); in GenAnalysisGraph() local
407 prior_node->pres.insert(input); in GenAnalysisGraph()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | cache_embedding.cc | 419 void CreateControlDepend(const FuncGraphPtr &main_graph, const AnfNodePtr &prior_node, const AnfNod… in CreateControlDepend() argument
424 AnfNodePtrList cd_inputs = {NewValueNode(prim::kPrimDepend), behind_node, prior_node}; in CreateControlDepend()
|