Home
last modified time | relevance | path

Searched defs:depend_inputs (Results 1 – 25 of 25) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Doverlap_recompute_allgather_and_flashattention_grad.cc78 …std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimDepend), grad_rs->input(kIndex1), re… in AddDependForRecomputedAllGatherAndGradientReduceScatter() local
159 std::vector<AnfNodePtr> depend_inputs{ in OverlapRecomputeAGAndFlashAttentionGrad() local
Dmerge_cast_opt.cc124 …std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimDepend), pair.first, new_make_tuple_… in MergeCastOpt() local
Doverlap_opt_shard_in_pipeline.cc137 std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimDepend), first_receive_cnode, in OverlapOptShardInPipeline() local
Dslice_activation_in_cell_share_recompute.cc64 std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimDepend), src_node, rely_node}; in CreateDependNode() local
Dsplit_layernorm_comm_fp.cc176 …std::vector<AnfNodePtr> depend_inputs = {NewValueNode(prim::kPrimDepend), post_cnode->input(kIndex… in InsertDependAndSetAbstract() local
Dcomp_comm_scheduling.cc1044 std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimDepend), input_node, source}; in AddRealDependencies() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/
Dsplit_umonad.cc47 …AnfNodePtrList depend_inputs = {NewValueNode(prim::kPrimDepend), original_inputs[input_idx], origi… in ProcessNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Dinplace_assign_for_custom_op.cc61 AnfNodePtrList depend_inputs = {NewValueNode(prim::kPrimDepend), src, dst}; in InplaceAssign() local
Dcommunication_op_fusion.cc591 …std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimDepend), cnode->input(1)->cast<CNode… in DoFusion() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dfloatstatus_addn_fusion.cc115 …AnfNodePtrList depend_inputs = {NewValueNode(prim::kPrimDepend), broadcast_to_node, maketuple_node… in ProcessFloatStatusAddN() local
Doptimize_assign.cc102 AnfNodePtrList depend_inputs = {NewValueNode(prim::kPrimDepend), assign_to_node, getitem}; in KeepExecOrder() local
Dinplace_assign_builder.cc272 …AnfNodePtrList depend_inputs = {NewValueNode(prim::kPrimDepend), iter.inplace_assignee_addr, iter.… in ProcessOriginCNodeUser() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/
Dslice_activation_in_recompute.cc176 …std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimDepend), forward_node_user.first, sl… in InsertSliceAllGatherNode() local
188 …std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimDepend), slice_allgathers[i - 1]->in… in InsertAllGatherDepend() local
Dgrouped_pairwise_exchange_alltoall.cc488 std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimDepend), prev_node, add_node}; in InsertDependOnBranches() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dfa_alltoallv_parallel.cc123 …std::vector<AnfNodePtr> depend_inputs{NewValueNode(std::make_shared<Primitive>(prim::kPrimDepend->… in CreateDepend() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dgrad_accumulation_utils.cc253 …std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimDepend), forward_start_node->input(1… in ReorderGradAccumulation() local
Dgraph_splitter.cc596 …std::vector<AnfNodePtr> depend_inputs = {NewValueNode(prim::kPrimDepend), replaced_output, make_tu… in HandleHungNodes() local
1649 …AnfNodePtrList depend_inputs = {NewValueNode(prim::kPrimDepend), func_graph_->output(), tuple_of_c… in AddControlEdgeForProcessWithoutIndegree() local
2056 …std::vector<AnfNodePtr> depend_inputs = {NewValueNode(prim::kPrimDepend), replaced_output, make_tu… in AddDependencyBetweenSegments() local
2226 std::vector<AnfNodePtr> depend_inputs = {NewValueNode(prim::kPrimDepend), replaced_output, in AddDependencyForSend() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dgpipe_interleave_scheduler.cc328 …std::vector<AnfNodePtr> depend_inputs = {NewValueNode(prim::kPrimDepend), begin_node->input(1), ma… in OptimizerShardCommReorder() local
Dpipeline_scheduler.cc465 …std::vector<AnfNodePtr> depend_inputs = {NewValueNode(prim::kPrimDepend), begin_node->input(1), ma… in OptimizerShardCommReorder() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dspecial_op_eliminate.h669 auto &depend_inputs = inputs[0]->cast<CNodePtr>()->inputs(); in operator() local
Drecompute.cc618 std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimMakeTuple)}; in AddDependNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Drecompute_helper.cc587 std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimDepend), first_input, make_tuple}; in NewRecomputedNode() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc99 std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimDepend), add, add_second}; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dcontrol_node_parser.cc1350 …std::vector<AnfNodePtr> depend_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimDepen… in InsertDependForParallelCall() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/parse/
Dparse.cc4543 …std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimDepend), context_expr_node, entered_… in ParseWith() local