| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | updatestate_eliminate.cc | 218 if (IsPrimitiveCNode(load_attach, prim::kPrimUpdateState)) { in GetLoadsFollowLoad()
247 if (IsPrimitiveCNode(prev_node, prim::kPrimUpdateState)) { in GetLoadsFollowTuple()
378 auto update_state = NewValueNode(prim::kPrimUpdateState); in EliminateUpdateStateForLoads()
419 auto new_update_state = fg->NewCNode({NewValueNode(prim::kPrimUpdateState), u1, make_tuple}); in EliminateUpdateStateBetweenAssigns()
514 … auto new_update_state = fg->NewCNode({NewValueNode(prim::kPrimUpdateState), u1, new_make_tuple}); in EliminateUpdateStateBetweenAssignMakeTuple()
555 if (!IsPrimitiveCNode(u2, prim::kPrimUpdateState) || !OnlyUsedByTwoNode(u2, a2, a3)) { in EliminateUpdateStateBetweenMakeTupleAssign()
588 … auto new_update_state = fg->NewCNode({NewValueNode(prim::kPrimUpdateState), u1, new_make_tuple}); in EliminateUpdateStateBetweenMakeTupleAssign()
613 if (IsPrimitiveCNode(u, prim::kPrimUpdateState) && OnlyUsedByTwoNode(u, assign, update_state)) { in EliminateUpdateStateForAssign()
698 if (IsPrimitiveCNode(node, prim::kPrimUpdateState)) { in operator ()()
732 if (IsPrimitiveCNode(node, prim::kPrimUpdateState)) { in operator ()()
[all …]
|
| D | item_tuple_or_list_eliminate.h | 415 if (!IsPrimitiveCNode(y_, prim::kPrimUpdateState)) { in ExistUpdateStateUser()
428 return IsPrimitiveCNode(user.first, prim::kPrimUpdateState); in ExistUpdateStateUser()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | adam_fusion.cc | 76 if (IsPrimitiveCNode(user.first, prim::kPrimUpdateState)) { in RelpaceOutputEdge()
107 VectorRef next_state = VectorRef({prim::kPrimUpdateState, u_, tuple_load}); in DefinePattern()
110 next_state = VectorRef({prim::kPrimUpdateState, next_state, assign_param}); in DefinePattern()
114 next_state = VectorRef({prim::kPrimUpdateState, next_state, assign_m}); in DefinePattern()
|
| D | adam_weight_decay_fusion.cc | 75 if (IsPrimitiveCNode(user.first, prim::kPrimUpdateState)) { in ReplaceOutputEdge()
109 VectorRef next_state = VectorRef({prim::kPrimUpdateState, u_, tuple_load}); in DefinePattern()
112 next_state = VectorRef({prim::kPrimUpdateState, next_state, assign_param}); in DefinePattern()
116 next_state = VectorRef({prim::kPrimUpdateState, next_state, assign_m}); in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | order_enforce.cc | 40 if (IsPrimitiveCNode(node, prim::kPrimUpdateState)) { in Run()
78 if (!IsPrimitiveCNode(prev_u, prim::kPrimUpdateState)) { in HandleUpdateState()
108 if (IsPrimitiveCNode(user_node, prim::kPrimUpdateState)) { in FindUpdateStateUsers()
246 …iveCNode(load_user, prim::kPrimMakeTuple) || IsPrimitiveCNode(load_user, prim::kPrimUpdateState)) { in AddInputEdges()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | auto_monad_eliminate.cc | 190 if (IsPrimitiveCNode(load_user.first, prim::kPrimUpdateState)) { in ReplaceLoadUser()
200 …etuple_users.size() == 1 && IsPrimitiveCNode(maketuple_users.back().first, prim::kPrimUpdateState); in ReplaceLoadUser()
270 if (!IsPrimitiveCNode(update_state, prim::kPrimUpdateState) || node_users.size() == kUserSize) { in ReplaceUpdateStateForLoad()
337 if (!IsPrimitiveCNode(attach, prim::kPrimUpdateState)) { in EliminateAutoMonadNode()
|
| D | irpass.cc | 191 prim::kPrimUpdateState); in OptimizeIRPassLib()
193 … "updatestate_pure_node_eliminater", prim::kPrimUpdateState); in OptimizeIRPassLib()
|
| D | recompute.cc | 125 if (IsPrimitiveCNode(current_node, prim::kPrimUpdateState)) { in GetMaxSubGraph()
376 if (IsPrimitiveCNode(input_cnode, prim::kPrimUpdateState)) { in NewRecomputedNode()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | pynative_dfunctor.cc | 41 if (IsPrimitiveCNode(cnode_morph, prim::kPrimUpdateState)) { in GenNewTensor()
172 …node_morph->size() - 1 != paras.size() && !IsPrimitiveCNode(cnode_morph, prim::kPrimUpdateState)) { in RunInputReplace()
196 if (para_ref_size > 0 && !IsPrimitiveCNode(input_node, prim::kPrimUpdateState)) { in RunInputReplace()
|
| D | dfunctor.cc | 525 …(prim->Hash() == prim::kPrimUpdateState->Hash() && prim->name() == prim::kPrimUpdateState->name())… in MapPrimitiveToK()
731 …rimitiveCNode(cnode, prim::kPrimStopGradient) || IsPrimitiveCNode(cnode, prim::kPrimUpdateState) || in BroadCastStopFlag()
882 if (IsPrimitiveCNode(user_node, prim::kPrimUpdateState)) { in RemovePrimalUpdateStates()
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/acl_vm/ |
| D | acl_multi_graph_session.cc | 104 …out, 0, false, std::vector<PrimitivePtr>{prim::kPrimMakeTuple, prim::kPrimUpdateState, prim::kPrim… in ConstructOutputRef()
135 … std::vector<PrimitivePtr>{prim::kPrimMakeTuple, prim::kPrimUpdateState, prim::kPrimStateSetItem}); in ConstructOutputRefByTupleNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | update_state_formatter.cc | 34 … [](const AnfNodePtr &node) { return IsPrimitiveCNode(node, prim::kPrimUpdateState); }); in GetUpdateStateList()
167 return IsPrimitiveCNode(user.first, prim::kPrimUpdateState); in FindIndexesToUpdateState()
247 if (IsPrimitiveCNode(user.first, prim::kPrimUpdateState)) { in ProcessIndex()
|
| D | optimize_assign.cc | 99 AnfNodePtrList update_state_inputs = {NewValueNode(prim::kPrimUpdateState), u, getitem}; in KeepExecOrder()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | optimize_updatestate.cc | 32 return VectorRef({prim::kPrimUpdateState, Xs}); in DefinePattern()
|
| D | convert_tuple_output_to_maketuple.cc | 74 if (IsPrimitiveCNode(cnode, prim::kPrimUpdateState)) { in Process()
|
| D | communication_op_fusion.cc | 266 if (IsPrimitiveCNode(make_tuple_user.first, prim::kPrimUpdateState)) { in AdjustAllReduceInputWithLoad()
279 if (IsPrimitiveCNode(load_user.first, prim::kPrimUpdateState)) { in AdjustAllReduceInputWithLoad()
|
| D | eliminate_redundant_op.cc | 62 } else if (IsPrimitive(input0, prim::kPrimUpdateState)) { in GetRealPrevCNode()
|
| D | optimize_dependence.cc | 161 if (!cnode->IsApply(prim::kPrimDepend) && !cnode->IsApply(prim::kPrimUpdateState)) { in SearchTransDataAndCast()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | anf.cc | 413 … (IsPrimitiveCNode(node, prim::kPrimMakeTuple) || IsPrimitiveCNode(node, prim::kPrimUpdateState)) { in GetVirtualNodeTargetFromInputs()
417 …if (IsPrimitiveCNode(node, prim::kPrimUpdateState) && inputs.size() > update_state_valid_input_ind… in GetVirtualNodeTargetFromInputs()
532 … IsPrimitiveCNode(node, prim::kPrimUpdateState) || IsPrimitiveCNode(node, prim::kPrimMakeTuple) || in GetOriginNodeTarget()
538 … IsPrimitiveCNode(node, prim::kPrimUpdateState) || IsPrimitiveCNode(node, prim::kPrimMakeTuple) || in GetOriginNodeTarget()
|
| /third_party/mindspore/mindspore/lite/tools/converter/import/ |
| D | mindspore_importer.cc | 132 if (opt::CheckPrimitiveType(output_node, prim::kPrimUpdateState) || in GetFuncGraphOutputName()
145 if (opt::CheckPrimitiveType(output_node, prim::kPrimUpdateState) || in GetFuncGraphOutputName()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | fusion_base_pass.cc | 76 if (!AnfAlgo::CheckPrimitiveType(user_node, prim::kPrimUpdateState)) { in GetNotUpdateStateUserNums()
|
| D | ub_pattern_fusion.cc | 302 if (AnfAlgo::CheckPrimitiveType(use_node.first, prim::kPrimUpdateState)) { in GetFusionScopeOutputNodeList()
318 if (AnfAlgo::CheckPrimitiveType(user.first, prim::kPrimUpdateState)) { in GetFusionScopeOutputNodeList()
|
| /third_party/mindspore/tests/ut/cpp/ir/ |
| D | manager_test.cc | 637 auto u1 = fg->NewCNode({NewValueNode(prim::kPrimUpdateState), u, x1}); in TEST_F()
662 auto u2 = fg->NewCNode({NewValueNode(prim::kPrimUpdateState), u, x1}); in TEST_F()
695 auto u1 = fg->NewCNode({NewValueNode(prim::kPrimUpdateState), u, x1}); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | helper.cc | 409 (cnode_name == prim::kPrimUpdateState->name())) { in GetRealNodeUsedList()
432 (cnode_name == prim::kPrimUpdateState->name())) { in GetRealNodeUsedListByOutputIdx()
930 AnfAlgo::CheckPrimitiveType(output, prim::kPrimUpdateState)) { in TransferDependOrUpdateState()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | convert.cc | 1012 if ((name == prim::kPrimDepend->name()) || (name == prim::kPrimUpdateState->name()) || in IsSourceEdgeNode()
1052 (name == prim::kPrimUpdateState->name()) || (name == prim::kPrimMakeTuple->name()) || in IsControlEdgeNode()
1098 if (name == prim::kPrimUpdateState->name()) { in AddEdgeForLoad()
1263 if (IsValueNode<None>(pred) || IsPrimitiveCNode(pred, prim::kPrimUpdateState)) { in GetRealInputNode()
1425 IsPrimitiveCNode(node, prim::kPrimUpdateState)) { in Convert()
|