Home
last modified time | relevance | path

Searched defs:node_user (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dfunctionalize_cond.cc42 auto node_user = node_users.front(); in GetSwitchBranchType() local
136 for (auto &node_user : node_users) { in IdentifySubgraphInput() local
263 for (auto &node_user : node_users) { in Process() local
Dfunctionalize_while.cc218 for (auto &node_user : node_users) { in UpdateExitNodeUser() local
356 for (auto &node_user : node_users) { in IdentifyCondSubgraphInput() local
496 for (auto &node_user : node_users) { in IdentifyBodySubgraphInput() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dreduce_same_act_pass.cc51 for (const auto &node_user : cur_node_users) { in Run() local
Dslice_prepose_pass.cc210 for (auto &node_user : node_users) { in SwapSliceWithPreceed() local
216 for (auto &node_user : node_users) { in SwapSliceWithPreceed() local
796 for (auto &node_user : node_users) { in PreposeWithAbnormalReshape() local
1064 for (auto &node_user : node_users) { in PreposeWithMatmul() local
1166 for (auto &node_user : node_users) { in PreposeWithFullConnection() local
1330 for (auto &node_user : node_users) { in PreposeWithArithmetic() local
1419 for (auto &node_user : node_users) { in MergeSequentialSlice() local
1464 for (auto &node_user : node_users) { in MergeParallelSlice() local
Ddecrease_transpose_algo.cc216 for (auto &node_user : node_users) { in PostTransposeFusion() local
376 for (auto &node_user : node_users) { in InsertPostTransNode() local
Dinfershape_pass.cc80 for (auto &node_user : node_users) { in ModifySubGraphInputCNodeFormat() local
Dredundant_op_remove_pass.cc245 for (auto &node_user : node_users) { in RemoveDropoutOp() local
Dtranspose_strategy.cc55 [](const std::pair<AnfNodePtr, int> &node_user) { return node_user.first; }); in GetPostNodes()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dset_fracz_group_attr.cc110 auto node_user = manager->node_users(); in GetCNodeNeighborFraczNodes() local
153 auto node_user = manager->node_users(); in GetNeighborFraczNodes() local
/third_party/mindspore/mindspore/lite/tools/optimizer/format/
Ddelete_redundant_transpose.cc153 for (auto &node_user : node_users) { in UpdateNodeFormat() local
Dto_format_base.cc149 for (auto &node_user : node_users) { in InsertPostTransNode() local
189 for (auto &node_user : node_users) { in DecideWhetherHandleGraphInput() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dparameter_manager.cc112 for (auto &node_user : load_node_users) { in FindParameterNodeUsers() local
176 for (auto node_user : node_users) { in IsUsedParameter() local
401 static void InsertFullySplitParamGradAccu(const std::pair<AnfNodePtr, int> &node_user, in InsertFullySplitParamGradAccu()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc111 void InsertVirtualAssignAdd(const std::pair<AnfNodePtr, int> &node_user, const FuncGraphManagerPtr … in InsertVirtualAssignAdd()
191 for (auto &node_user : node_set) { in HandleReceiveParam() local
221 for (auto &node_user : node_set) { in AddVirtualAssignAdd() local
430 for (auto &node_user : node_users) { in BroadCastNeedGrad() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtflite_lstm_cell_fusion.cc336 for (auto &node_user : while_node_users) { in CheckReferencedOutputs() local
640 for (auto &node_user : while_node_users) { in AdjustOtherGetItems() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc157 for (auto &node_user : node_users) { in LabelMicroBatch() local
388 for (auto &node_user : node_users) { in GetParameterPair() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_select_graph_kernel.cc366 for (auto &node_user : iter->second) { in CheckFormatsAndDtypes() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc572 for (auto &node_user : node_users) { in FindCNodeOutput() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.cc390 for (auto &node_user : node_users) { in AddCommOpParamFlag() local