Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dsplit_umonad.cc51 auto depend_cnode = func_graph->NewCNode(depend_inputs); in ProcessNode() local
52 depend_cnode->set_abstract(original_inputs[input_idx]->abstract()); in ProcessNode()
53 depend_cnode->set_kernel_info(std::make_shared<device::KernelInfo>()); in ProcessNode()
58 new_inputs.push_back(depend_cnode); in ProcessNode()
Dadd_atomic_clean.cc373 auto depend_cnode = main_graph->NewCNode(d_inputs); in AddDepend() local
374 depend_cnode->set_abstract(clean_node->abstract()); in AddDepend()
375 main_graph->AddNode(depend_cnode); in AddDepend()
379 user_cnode->set_input(IntToSize(index), depend_cnode); in AddDepend()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dadjust_depend_for_parallel_optimizer_recompute_all_gather.cc107 auto depend_cnode = depend_node->cast<CNodePtr>(); in AdjustAllgatherDepend() local
116 … allgather_next_node, AnfAlgo::GetInputNode(depend_cnode, 1)}; in AdjustAllgatherDepend()
119 manager->SetEdge(node, 1, AnfAlgo::GetInputNode(depend_cnode, 0)); in AdjustAllgatherDepend()
Doptimize_dependence.cc216 auto depend_cnode = node->cast<CNodePtr>(); in GetConvertNode() local
217 MS_EXCEPTION_IF_NULL(depend_cnode); in GetConvertNode()
218 auto replacing_node = depend_cnode->input(index); in GetConvertNode()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Ditem_tuple_or_list_eliminate.h377 auto depend_cnode = depend->cast<CNodePtr>(); in operator() local
382 inputs[1] = depend_cnode->input(1); in operator()
389 auto depend_node = NewCNode({depend_cnode->input(0), item_node, y_}, fg); in operator()
Dupdatestate_eliminate.cc317 const auto &depend_cnode = depend_node->cast<CNodePtr>(); in EliminateUselessNodesForUpdateStates() local
318 mgr->Replace(depend_cnode, depend_cnode->input(kInputIndex)); in EliminateUselessNodesForUpdateStates()
Dbranch_culling.cc282 auto depend_cnode = graph->NewCNode(inputs); in GenerateSwitchDependNode() local
283 if (!manager->Replace(square_op, depend_cnode)) { in GenerateSwitchDependNode()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc425 auto depend_cnode = main_graph->NewCNode(cd_inputs); in CreateControlDepend() local
426 if (!manager->Replace(behind_node, depend_cnode)) { in CreateControlDepend()
438 auto depend_cnode = graph->NewCNode(depend_list); in CreateDepend() local
439 depend_cnode->set_abstract(patron_node->abstract()); in CreateDepend()
440 return depend_cnode; in CreateDepend()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_auto_monad.cc1306 auto depend_cnode = kernel_graph_->NewCNode({depend, origin, input}); in MakeDepend() local
1307 depend_cnode->set_abstract(origin->abstract()); in MakeDepend()
1308 return depend_cnode; in MakeDepend()
1317 auto depend_cnode = MakeDepend(origin_output, monad); in MakeMonadDepend() local
1318 kernel_graph_->set_output(depend_cnode); in MakeMonadDepend()
Dsession_basic.cc1916 auto depend_cnode = user.first->cast<CNodePtr>(); in ExtendNodeUsers() local
1917 if (depend_cnode == nullptr) { in ExtendNodeUsers()
1920 if (front_node != depend_cnode->input(1)) { in ExtendNodeUsers()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.cc135 …auto depend_cnode = CheckAnfNodeIfCNodeAndInputSize(transop_cnode->input(1), kDependInputTensorNum… in EliminateDependTransop() local
136 …auto prev_transop_cnode = CheckAnfNodeIfCNodeAndInputSize(depend_cnode->input(1), kTransOpInputTen… in EliminateDependTransop()
141 depend_cnode->input(kDependAttachNodeIndex)}; in EliminateDependTransop()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Dauto_monad.cc1389 auto depend_cnode = func_graph_->NewCNode({depend, output, node}); in AttachToOutput() local
1390 depend_cnode->set_abstract(output->abstract()); in AttachToOutput()
1391 func_graph_->set_output(depend_cnode); in AttachToOutput()