Home
last modified time | relevance | path

Searched defs:depend_input (Results 1 – 17 of 17) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dconvert_data_depend_to_control_depend.cc50 …std::vector<AnfNodePtr> depend_input = {NewValueNode(std::make_shared<Primitive>(kDependOpName)), … in Process() local
Dadjust_print_for_ge.cc140 …std::vector<AnfNodePtr> depend_input = {NewValueNode(std::make_shared<Primitive>(kDependOpName)), … in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dsend_op_add_control_depend.cc65 …std::vector<AnfNodePtr> depend_input = {NewValueNode(std::make_shared<Primitive>(kDependOpName)), … in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/
Dinsert_depend_for_all_reduce.cc32 …std::vector<AnfNodePtr> depend_input = {NewValueNode(prim::kPrimDepend), post_cnode->input(1), pri… in InsertDepend() local
Dinsert_depend_for_all_gather_output.cc36 …std::vector<AnfNodePtr> depend_input = {NewValueNode(prim::kPrimDepend), post_cnode->input(1), pri… in InsertDepend() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Dreorder_send_recv_between_fp_bp.cc71 …std::vector<AnfNodePtr> depend_input = {NewValueNode(prim::kPrimDepend), input_comm_cnode, user_co… in InsertReorderDepend() local
Doverlap_opt_shard_in_pipeline.cc211 … std::vector<AnfNodePtr> depend_input = {NewValueNode(prim::kPrimDepend), rs->input(1), last_send}; in OverlapOptShardGradInPipeline() local
Doverlap_recompute_and_grad_model_parallel.cc164 …std::vector<AnfNodePtr> depend_input = {NewValueNode(prim::kPrimDepend), post_cnode->input(index),… in InsertDepend() local
590 …std::vector<AnfNodePtr> depend_input = {NewValueNode(prim::kPrimDepend), dst_node_input, new_src_n… in OverlapRecomputeAndGradModelParallel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dneighbor_exchange_v2_fusion.cc440 …std::vector<AnfNodePtr> depend_input = {NewValueNode(std::make_shared<Primitive>(prim::kPrimDepend… in CreateAllToAllvNode() local
682 …std::vector<AnfNodePtr> depend_input = {NewValueNode(std::make_shared<Primitive>(prim::kPrimDepend… in AllToAllvRecvEmpty() local
929 …std::vector<AnfNodePtr> depend_input = {NewValueNode(std::make_shared<Primitive>(prim::kPrimDepend… in CreateSplitGradNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Dneighbor_exchange_v2_unify_mindir.cc376 …std::vector<AnfNodePtr> depend_input = {NewValueNode(std::make_shared<Primitive>(prim::kPrimDepend… in CreateAllToAllvNode() local
613 …std::vector<AnfNodePtr> depend_input = {NewValueNode(std::make_shared<Primitive>(prim::kPrimDepend… in AllToAllvRecvEmpty() local
866 …std::vector<AnfNodePtr> depend_input = {NewValueNode(std::make_shared<Primitive>(prim::kPrimDepend… in CreateSplitGradNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Drandom_ops_proto.cc337 std::vector<std::string> depend_input{"shape"}; in CUST_IMPLEMT_INFERFUNC() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_interleave.cc1221 std::vector<AnfNodePtr> depend_input = {NewValueNode(prim::kPrimDepend), out, temp}; in FreezeGradient() local
1235 … std::vector<AnfNodePtr> depend_input = {NewValueNode(prim::kPrimDepend), cnode->input(1), grads}; in FreezeGradient() local
Dpipeline_scheduler.cc206 …std::vector<AnfNodePtr> depend_input = {NewValueNode(prim::kPrimDepend), node_last->input(1), node… in ControlOrder() local
Dpipeline_transformer.cc1953 std::vector<AnfNodePtr> depend_input = {NewValueNode(prim::kPrimDepend), out, temp}; in FreezeGradient() local
1967 … std::vector<AnfNodePtr> depend_input = {NewValueNode(prim::kPrimDepend), cnode->input(1), grads}; in FreezeGradient() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dupdatestate_eliminate.cc177 auto depend_input = depend->input(kInputIndex); in EliminateUpdateStateWithDepend() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc686 …std::vector<AnfNodePtr> depend_input = {NewValueNode(prim::kPrimDepend), post_cnode->input(1), pri… in InsertDepend() local
Dgraph_splitter.cc2065 …std::vector<AnfNodePtr> depend_input = {NewValueNode(prim::kPrimDepend), recv->cast<CNodePtr>()->i… in AddDependencyBetweenSegments() local