Home
last modified time | relevance | path

Searched defs:node_users (Results 1 – 25 of 109) sorted by relevance

12345

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tf/
Dfunctionalize_cond.cc40 auto node_users = manager->node_users()[switch_cnode]; in GetSwitchBranchType() local
167 auto node_users = manager->node_users()[cnode]; in IdentifySubgraphInput() local
280 auto node_users = manager->node_users()[switch_op]; in DegenerateNonControlFlow() local
352 auto node_users = manager->node_users()[merge_node_]; in Process() local
Dtf_fake_quant_adjust.cc75 auto node_users = manager->node_users()[cnode]; in RemoveFakeQuant() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/hcom/
Dinsert_tensor_move_for_hccl_op_ge.cc52 auto &node_users = manager->node_users()[cnode]; in IsCareNode() local
68 auto &node_users = manager->node_users(); in IsNodeOutPutUsedByOtherRealKernel() local
100 const auto &node_users = manager->node_users(); in AdjustDepend() local
Dinsert_load_for_allgather.cc61 auto node_users = mng->node_users()[node]; in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/format/
Ddelete_redundant_transpose.cc148 auto node_users = manager_->node_users()[cnode]; in DoTransTransFusion() local
227 auto node_users = manager_->node_users()[cnode]; in UpdateNodeFormat() local
269 const AnfNodeIndexSet &node_users) { in CopyQuantParam()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/
Dstorage_format_convertor.cc38 AnfNodePtr GetUsedOperator(const AnfNodePtr &node, const NodeUsersMap &node_users, const PrimitiveP… in GetUsedOperator()
50 bool IsUsedByConv2D(const AnfNodePtr &node, const NodeUsersMap &node_users) { in IsUsedByConv2D()
65 bool IsUsedBySwitch(const AnfNodePtr &node, const NodeUsersMap &node_users) { in IsUsedBySwitch()
119 AnfNodePtr GetMomentumVarByAccum(const AnfNodePtr &node, const NodeUsersMap &node_users) { in GetMomentumVarByAccum()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/
Donnx_custom_op_adjust.cc93 auto node_users = manager->node_users()[cnode]; in AdjustRot90() local
129 auto node_users = manager->node_users()[cnode]; in AdjustRandomUniformLike() local
166 auto node_users = manager->node_users()[cnode]; in AdjustRandomUniformLike() local
Donnx_quantize_linear_adjust.cc41 auto node_users = manager->node_users()[cnode]; in RemoveDequantizeLinear() local
123 auto node_users = manager->node_users()[cnode]; in FoldQuantParms() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dkvcache_quant_pass.cc122 auto node_users = manager->node_users()[cast_cnode]; in ReplaceCastOpToQuantOp() local
147 auto node_users = manager->node_users()[cast_cnode]; in ReplaceCastOpToAntiQuantOp() local
168 auto node_users = manager->node_users()[anf_node]; in RemoveOp() local
Dremove_load_pass.cc36 auto node_users = manager->node_users()[cnode]; in Run() local
Dspecify_graph_input_format.cc115 auto node_users = manager->node_users(); in CheckInputsFormatNHWC() local
148 auto node_users = manager->node_users(); in GetTracedCnodes() local
Dsplit_one_pass.cc53 auto node_users = manager->node_users()[cnode]; in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/common/
Dfunc_graph_subgraph.cc132 auto node_users = belong_anf_->manager()->node_users(); in GetOutputCNodes() local
157 auto node_users = belong_anf_->manager()->node_users(); in InitSubGraphNode() local
224 auto node_users = belong_anf_->manager()->node_users(); in InitSubGraphInNode() local
264 auto node_users = belong_anf_->manager()->node_users(); in InitSubGraphOutNode() local
516 auto node_users = this->belong_anf_->manager()->node_users(); in CreateParameterForPartialSubGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/
Dinsert_tensor_move_for_hccl_op.cc41 auto &node_users = manager->node_users(); in IsNodeOutPutUsedByOtherRealKernel() local
121 const auto &node_users = manager->node_users(); in AdjustDependToTensorMove() local
Dinsert_depend_for_all_gather_output.cc108 auto node_users = graph->manager()->node_users()[node]; in GetEachSegSend() local
149 auto node_users = graph->manager()->node_users()[node]; in IsChanged() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/
Dslice_activation_in_recompute.cc77 auto node_users = manager->node_users()[node]; in IsDuplicateNode() local
88 auto node_users = manager->node_users()[root_node]; in GroupingNextNodes() local
112 …d InsertSliceAllGatherNode(const std::vector<std::pair<std::shared_ptr<AnfNode>, int>> &node_users, in InsertSliceAllGatherNode()
197 auto node_users = manager->node_users()[last_allgather]; in InsertAllGatherDepend() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dreshape_reduce_fusion.cc59 auto node_users = manager->node_users()[cnode]; in CheckIsClosedCycle() local
136 auto node_users = manager->node_users()[reduce]; in CheckReduceReshapeFusion() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Doptimize_updatestate.cc46 auto &node_users = manager->node_users(); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dremove_tensor_to_scalar_or_tuple_ops.cc68 const auto &node_users = manager->node_users(); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/core/utils/
Dparallel_node_check.cc95 auto node_users = manager->node_users()[cnode]; in IsReshapeBetweenDropoutGenMaskAndFlashAttentionScore() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Drecompute_helper.cc103 const auto &node_users = mng->node_users(); in FindCandidateRecomputedNodes() local
155 const auto &node_users = mng->node_users(); in GetMaxSubGraph() local
182 const auto &node_users = mng->node_users(); in GetOriginRecomputeAndTargetNodes() local
396 const auto &node_users = mng->node_users(); in HasForwardOutput() local
410 const auto &node_users = mng->node_users(); in GetTupleGetItemOutputNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dparameter_eliminate.h52 auto &node_users = manager->node_users(); in GetCallers() local
257 const auto &node_users = manager->node_users(); in AdjustGetItemCall() local
353 const auto &node_users = manager->node_users(); in operator() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/src/
Dgraph_split_api.cc86 auto node_users = manager->GetUsers(cur_cnode); in GetSubgraphTotalCNodes() local
196 auto node_users = manager->GetUsers(cnode); in FilterMakeTuple() local
351 auto node_users = manager->GetUsers(cnode); in GetSubgraphOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/
Dremove_internal_output.cc31 auto &node_users = manager->node_users(); in UsedForOutputOnly() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/ad/
Dprim_bprop_optimizer.cc65 const auto &node_users = manager->node_users(); in AnalysisArgUsingInfo() local
77 void PrimBpropOptGraphLevel2Info::AnalysisNodeUsingInfo(const NodeUsersMap &node_users, in AnalysisNodeUsingInfo()
108 void PrimBpropOptGraphLevel2Info::AalysisForTupleGetItem(const NodeUsersMap &node_users, in AalysisForTupleGetItem()

12345