Home
last modified time | relevance | path

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

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/
Dstorage_format_operator.cc61 std::optional<std::string> GetApplyMomentumFormat(const AnfNodePtr &user_node, in GetApplyMomentumFormat()
87 std::optional<std::string> GetBNTrainingUpdateFormat(const AnfNodePtr &user_node, in GetBNTrainingUpdateFormat()
Dstorage_format_convertor.cc209 for (const auto &user_node : output_nodes) { in SetStorageFormatFromConfig() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/static_analysis/
Dorder_enforce.cc107 auto side_effect_nodes = FindNodeUsers(prev_u, [&update_state](const AnfNodePtr &user_node) { in HandleUpdateState()
134 auto &user_node = user.first; in FindUpdateStateUsers() local
196 auto maketuple_users = FindNodeUsers(maketuple, [](const AnfNodePtr &user_node) { in HandleMakeTupleUsers()
370 auto &user_node = user.first; in FindNodeUsers() local
387 auto &user_node = user.first; in FindParallelNodeUsers() local
401 return FindNodeUsers(load_or_param, [this](const AnfNodePtr &user_node) { in FindLoadUsers()
409 return FindNodeUsers(param, [this](const AnfNodePtr &user_node) { in FindLoadNodes()
501 auto &user_node = user.first; in ReCheckUsersOfLoad() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Ddropout_gen_mask_fusion.cc44 auto user_node = user_pair.first->cast<CNodePtr>(); in DoFusion() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Dspecialized_prepare.cc32 auto user_node = iter.first; in InsertFlowOutputToTA() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Draise_reduction_precision.cc71 auto user_node = user.first; in ReplaceNode() local
Dadd_stitch_atomic_clean_gpu.cc58 … const AnfNodePtr &composite_node, const AnfNodePtr &user_node, in AddDepend()
Ddecrease_transfer_precision.cc260 auto user_node = user.first; in ProcessSon() local
Dinplace_assign_builder.cc239 const auto &user_node = node_index.first; in FindOriginCNodeUsers() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dparameter_eliminate.h65 auto &user_node = user.first; in GetCallers() local
399 auto user_node = node_and_idx.first->first; in GetUserPartialNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/
Dshape_ops_splitter.cc55 auto user_node = user->cast<CNodePtr>(); in SplitNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_interleave.cc333 auto user_node = user_pair.first->cast<CNodePtr>(); in Coloring() local
367 auto user_node = user_pair.first->cast<CNodePtr>(); in BroadCastColoring() local
522 for (auto &user_node : node_users) { in RemoveMonadNode() local
550 void PipelineInterleave::InsertSendReceive(const AnfNodePtr &node, const AnfNodePtr &user_node, int… in InsertSendReceive()
607 auto user_node = user_pair.first; in CutBorderForNode() local
1150 auto user_node = user.first; in HandleSendParam() local
1211 auto user_node = grad_user.first->cast<CNodePtr>(); in FreezeGradient() local
Dfold_pipeline_transformer.cc126 auto user_node = user_pair.first->cast<CNodePtr>(); in Coloring() local
194 auto user_node = user_pair.first->cast<CNodePtr>(); in BroadCastColoring() local
505 auto user_node = user_pair.first; in CutBorderForNode() local
Dpipeline_transformer.cc502 auto user_node = user_pair.first->cast<CNodePtr>(); in Coloring() local
534 auto user_node = user_pair.first->cast<CNodePtr>(); in BroadCastColoring() local
933 for (auto &user_node : node_users) { in RemoveMonadNode() local
1227 auto user_node = user_pair.first; in CutBorderForNode() local
1943 auto user_node = grad_user.first->cast<CNodePtr>(); in FreezeGradient() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/
Dvalidator.cc203 auto user_node = user.first; in CheckAssignReturnValue() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dbatch_norm_add_relu_fusion.cc134 for (auto user_node : user_nodes) { in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dgraph_splitter.cc482 auto user_node = user.first; in FindNextUpdateStateNode() local
1528 const auto &user_node = user.first; in AddDataSyncNode() local
1732 auto user_node = user_pair.first; in EliminateDataSyncNode() local
1773 auto user_node = std::get<2>(send_recv_pair); in DumpDistributedGraph() local
1917 auto user_node = u.first->cast<CNodePtr>(); in FindInterProcessOutDegree() local
2085 auto user_node = std::get<2>(send_recv_pair); in EliminateExtraNodes() local
2119 const auto &user_node = std::get<3>(send_recv_pair); in ReplaceOriginNodesWithRecv() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dunfold_nested_output.cc188 auto user_node = user.first; in ProcessSucceedNode() local
Dunfold_maketuple.cc224 auto user_node = user.first; in ProcessSucceedNodes() local
Dbatchnorm_transform.cc222 auto user_node = iter.first; in TransformBatchNorm() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dscalar_op_pass.cc87 size_t ScalarOpPass::GetInputNodeIndex(const AnfNodePtr &input, const CNodePtr &user_node) { in GetInputNodeIndex()
457 auto user_node = tensor_to_scalar_user.first->cast<CNodePtr>(); in RemoveTensorToScalar() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/ad/
Dprim_bprop_optimizer.cc96 auto user_node = user_info.first; in AnalysisNodeUsingInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dcame_parallel_handler.cc122 auto user_node = item.first; in ForwardSearchCNode() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ir/
Dfunc_graph_cloner.cc722 const auto &user_node = cnode_index.first->first; in LiftParameters() local
737 const auto &user_node = cnode_index.first->first; in Lift() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc590 for (const auto &user_node : user_set) { in ReplaceNoRefToParams() local
601 for (const auto &user_node : indices_user_set) { in ReplaceNoRefToParams() local

12