Home
last modified time | relevance | path

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

12

/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dfunctionalize_cond.cc37 auto node_users = manager->node_users()[switch_cnode]; in GetSwitchBranchType() local
135 auto node_users = manager->node_users()[cnode]; in IdentifySubgraphInput() local
262 auto node_users = manager->node_users()[merge_node_]; in Process() local
Dfunctionalize_while.cc217 auto node_users = manager->node_users()[node]; in UpdateExitNodeUser() local
355 auto node_users = manager->node_users()[cnode]; in IdentifyCondSubgraphInput() local
495 auto node_users = manager->node_users()[cnode]; in IdentifyBodySubgraphInput() local
/third_party/mindspore/mindspore/lite/tools/common/
Dfunc_graph_subgraph.cc130 auto node_users = belong_anf_->manager()->node_users(); in GetOutputCNodes() local
155 auto node_users = belong_anf_->manager()->node_users(); in InitSubGraphNode() local
222 auto node_users = belong_anf_->manager()->node_users(); in InitSubGraphInNode() local
262 auto node_users = belong_anf_->manager()->node_users(); in InitSubGraphOutNode() local
512 auto node_users = this->belong_anf_->manager()->node_users(); in CreateParameterForPartialSubGraph() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Drecompute.cc91 const auto &node_users = mng->node_users(); in FindCandidateRecomputedNodes() local
142 const auto &node_users = mng->node_users(); in GetMaxSubGraph() local
169 const auto &node_users = mng->node_users(); in GetOriginRecomputeAndTargetNodes() local
243 const auto &node_users = mng->node_users(); in HasForwardOutput() local
257 const auto &node_users = mng->node_users(); in GetTupleGetItemOutputNodes() local
Dauto_monad_eliminate.cc268 auto &node_users = mgr->node_users()[update_state]; in ReplaceUpdateStateForLoad() local
340 auto &node_users = manager->node_users(); in EliminateAutoMonadNode() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Doptimize_updatestate.cc47 auto &node_users = manager->node_users(); in Process() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Dprim_bprop_optimizer.cc63 const auto &node_users = manager->node_users(); in AnalysisArgUsingInfo() local
75 void PrimBpropOptGraphLevel2Info::AnalysisNodeUsingInfo(const NodeUsersMap &node_users, in AnalysisNodeUsingInfo()
106 void PrimBpropOptGraphLevel2Info::AalysisForTupleGetItem(const NodeUsersMap &node_users, in AalysisForTupleGetItem()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dremove_internal_output.cc28 auto &node_users = manager->node_users(); in UsedForOutputOnly() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dmerge_addn.h123 auto &node_users = mng_->node_users(); in is_unique() local
Dincorporate_getitem.h386 const auto &node_users = it->second; in MultipleUse() local
471 auto &node_users = it->second; in TransformFuncGraph() local
804 auto &node_users = it->second; in MultipleUseOfSwitch() local
/third_party/mindspore/mindspore/lite/tools/optimizer/format/
Ddelete_redundant_transpose.cc152 auto node_users = manager_->node_users()[cnode]; in UpdateNodeFormat() local
Dto_format_base.cc148 auto node_users = manager_->node_users()[cnode]; in InsertPostTransNode() local
188 auto node_users = manager_->node_users()[input]; in DecideWhetherHandleGraphInput() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dredundant_op_remove_pass.cc96 auto node_users = manager->node_users()[pre_node]; in ProcessDependencyWithTwoNodes() local
244 auto node_users = manager->node_users()[anf_node]; in RemoveDropoutOp() local
Dslice_prepose_pass.cc208 auto node_users = manager->node_users()[slice_cnode]; in SwapSliceWithPreceed() local
795 auto node_users = manager->node_users()[slice_cnode]; in PreposeWithAbnormalReshape() local
988 auto node_users = manager->node_users()[slice_cnode]; in PreposeWithMatmul() local
1165 auto node_users = manager->node_users()[slice_cnode]; in PreposeWithFullConnection() local
1243 auto node_users = manager->node_users()[slice_cnode]; in PreposeWithArithmetic() local
1368 auto node_users = manager->node_users()[slice1_cnode]; in MergeSequentialSlice() local
1463 auto node_users = manager->node_users()[slices->at(i).first]; in MergeParallelSlice() local
Dtranspose_strategy.cc49 auto node_users = manager->node_users()[cnode]; in GetPostNodes() local
491 auto node_users = manager->node_users()[cnode]; in TransposeInsertDependOnShape() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dinsert_tensor_move_for_cascade.cc46 auto &node_users = manager->node_users(); in IsPartOutputsOfHcclOp() local
Dinsert_tensor_move_for_hccl_op.cc47 auto &node_users = manager->node_users(); in IsNodeOutPutUsedByOtherRealKernel() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc181 auto node_users = node_users_map[node]; in HandleReceiveParam() local
211 auto node_users = node_users_map[parameter]; in AddVirtualAssignAdd() local
414 auto node_users = (*node_users_map)[node]; in BroadCastMicroBatch() local
429 auto node_users = (*node_user_map)[node]; in BroadCastNeedGrad() local
529 auto node_users = (*node_users_map)[cnode]; in Micro() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc156 auto node_users = node_user_map[virtual_dataset_]; in LabelMicroBatch() local
214 auto node_users = manager_->node_users()[node]; in Coloring() local
239 auto node_users = manager_->node_users(); in BroadCastColoring() local
387 auto node_users = node_users_map[node]; in GetParameterPair() local
829 auto node_users = manager_->node_users()[node]; in CutBorder() local
/third_party/mindspore/mindspore/core/ir/
Dmanager.h316 NodeUsersMap &node_users() { return node_users_; } in node_users() function
318 const NodeUsersMap &node_users() const final { return node_users_; } in node_users() function
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Dorder_enforce.cc99 auto &node_users = manager_->node_users(); in FindUpdateStateUsers() local
313 auto &node_users = manager_->node_users(); in FindNodeUsers() local
/third_party/mindspore/mindspore/ccsrc/vm/
Dsegment_runner.cc60 auto &node_users = iter->second; in GetOutput() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_swap_manager.cc139 auto &node_users = iter->second; in CheckDistanceBetweenKernels() local
163 auto &node_users = iter->second; in CheckDistanceBetweenKernelsWithIdx() local
718 auto &node_users = item.second; in DumpUserNodes() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dadam_fusion.cc68 auto &node_users = mgr->node_users(); in RelpaceOutputEdge() local
Dadam_weight_decay_fusion.cc67 auto &node_users = mgr->node_users(); in ReplaceOutputEdge() local

12