Searched defs:update_state_node (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/ |
| D | lamb_fission.cc | 78 auto update_state_node = NewCNode(new_node_inputs, graph); in CreateUpdateStateNode() local
99 … const bool is_exist_umonad_node, const AnfNodePtr &update_state_node) { in CreateLambApplyOptimizerAssignNode()
174 const AnfNodePtr &update_state_node) { in CreateLambApplyWeightAssignNode()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/ |
| D | lamb_fission.cc | 78 auto update_state_node = NewCNode(new_node_inputs, graph); in CreateUpdateStateNode() local
107 … const bool is_exist_umonad_node, const AnfNodePtr &update_state_node) { in CreateLambApplyOptimizerAssignNode()
227 const AnfNodePtr &update_state_node) { in CreateLambApplyWeightAssignNode()
|
| D | batchnorm_transform.cc | 123 auto update_state_node = inputs[2]; in CreateAssignSubNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | incorporate_call.h | 138 AnfNodePtr update_state_node = nullptr; in MakeNewNode() local
|
| D | updatestate_eliminate.cc | 727 auto update_state_node = dyn_cast<CNode>(node); in operator ()() local
769 auto update_state_node = dyn_cast<CNode>(node); in operator ()() local
904 auto update_state_node = node->cast<CNodePtr>(); in operator ()() local
|
| D | vmap_eliminate.cc | 124 auto update_state_node = vmap_fg->NewCNode({update_state_prim, u_monad_node, attach_cnode}); in UpdateParam() local
155 auto update_state_node = UpdateParam(vmap_fg, u_monad_node, false, *param_mapping_table); in BindUMonad() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | graph_splitter.cc | 509 auto update_state_node = func_graph->NewCNode(inputs); in CreateUpdateStateNode() local
1450 CNodePtr update_state_node = FindNextUpdateStateNode(func_graph_, cnode); in ProcessRefNodes() local
1514 …GraphSplitter::AddDataSyncNode(const CNodePtr &side_effect_node, const CNodePtr &update_state_node, in AddDataSyncNode()
1713 CNodePtr update_state_node = CreateUpdateStateNode(func_graph_, update_state_inputs); in EliminateDataSyncNode() local
|