Home
last modified time | relevance | path

Searched defs:all_nodes (Results 1 – 23 of 23) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_auto_parallel.cc94 std::vector<AnfNodePtr> all_nodes = DeepScopedGraphSearch(ret); in StepAutoParallel() local
389 Status ConstructCostGraphNodesByUniqueId(const std::vector<AnfNodePtr> &all_nodes, const FuncGraphP… in ConstructCostGraphNodesByUniqueId()
504 Status ConstructCostGraphNodesByUniqueIdTC(const std::vector<AnfNodePtr> &all_nodes, const FuncGrap… in ConstructCostGraphNodesByUniqueIdTC()
651 void ConstructCostGraphEdges(const std::vector<AnfNodePtr> &all_nodes) { in ConstructCostGraphEdges()
731 void AugmentCostGraph(const std::vector<AnfNodePtr> &all_nodes) { in AugmentCostGraph()
851 void ReshapeCostCompute(const std::vector<AnfNodePtr> &all_nodes) { in ReshapeCostCompute()
906 Status ParallelStrategySearch(const std::vector<AnfNodePtr> &all_nodes, const FuncGraphPtr &root) { in ParallelStrategySearch()
1049 Status ParallelStrategyRecSearch(const std::vector<AnfNodePtr> &all_nodes, const FuncGraphPtr &root… in ParallelStrategyRecSearch()
Dstep_parallel.cc494 std::vector<AnfNodePtr> all_nodes = DeepScopedGraphSearch(ret); in HasStrategy() local
518 bool FindCommunicationOp(const std::vector<AnfNodePtr> &all_nodes) { in FindCommunicationOp()
860 void InsertVirtualOutput(const FuncGraphPtr &root, const std::vector<AnfNodePtr> &all_nodes) { in InsertVirtualOutput()
1875 void ExtractInformation(const std::vector<AnfNodePtr> &all_nodes) { in ExtractInformation()
2184 void ReshapeInit(const std::vector<AnfNodePtr> &all_nodes) { in ReshapeInit()
2544 void ParallelCommunication(const FuncGraphPtr &root, const std::vector<AnfNodePtr> &all_nodes, in ParallelCommunication()
2655 void CheckpointStrategy(const std::vector<AnfNodePtr> &all_nodes, const FuncGraphPtr &root) { in CheckpointStrategy()
2717 void SetForwardFlag(const std::vector<AnfNodePtr> &all_nodes) { in SetForwardFlag()
2734 void SetForwardFlag(const AnfNodeSet &all_nodes) { in SetForwardFlag()
2752 const auto &all_nodes = root->nodes(); in ForwardGraph() local
[all …]
Dparameter_manager.cc245 void CheckParameterSplit(const std::vector<AnfNodePtr> &all_nodes) { in CheckParameterSplit()
304 void HandleSymbolicKeyInstance(const FuncGraphPtr &root, const std::vector<AnfNodePtr> &all_nodes) { in HandleSymbolicKeyInstance()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dvalidator.cc133 AnfNodeSet &all_nodes = mgr->all_nodes(); in Validate() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dgraph_info.cc31 std::vector<AnfNodePtr> all_nodes = DeepScopedGraphSearch(ret); in FindPrimtive() local
Dpipeline_split_utils.cc164 void HandleReceiveParam(const FuncGraphPtr &root, const std::vector<AnfNodePtr> &all_nodes) { in HandleReceiveParam()
481 void LastStageEndNode(const std::vector<AnfNodePtr> &all_nodes, const FuncGraphManagerPtr &manager, in LastStageEndNode()
540 void ParameterStartNode(const std::vector<AnfNodePtr> &all_nodes, const FuncGraphManagerPtr &manage… in ParameterStartNode()
556 void HandleMicroBatch(const std::vector<AnfNodePtr> &all_nodes, const FuncGraphManagerPtr &manager)… in HandleMicroBatch()
/third_party/mindspore/tests/ut/cpp/parallel/
Dstep_parallel_test.cc371 std::vector<AnfNodePtr> all_nodes = DeepScopedGraphSearch(ret); in TEST_F() local
380 std::vector<AnfNodePtr> all_nodes = DeepScopedGraphSearch(ret); in TEST_F() local
389 std::vector<AnfNodePtr> all_nodes = DeepScopedGraphSearch(ret); in TEST_F() local
408 std::vector<AnfNodePtr> all_nodes = DeepScopedGraphSearch(ret); in TEST_F() local
447 std::vector<AnfNodePtr> all_nodes = DeepScopedGraphSearch(ret); in TEST_F() local
470 std::vector<AnfNodePtr> all_nodes = DeepScopedGraphSearch(ret); in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dbranch_culling.h106 std::vector<AnfNodePtr> all_nodes = DeepScopedGraphSearch(ret); in operator() local
Dgradient_eliminate.cc136 std::vector<AnfNodePtr> all_nodes = DeepScopedGraphSearch(ret); in GetJPrim() local
Dupdatestate_eliminate.cc713 auto &all_nodes = manager->all_nodes(); in operator ()() local
747 auto &all_nodes = manager->all_nodes(); in operator ()() local
783 auto &all_nodes = manager->all_nodes(); in operator ()() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dopt.cc198 auto &all_nodes = manager->all_nodes(); in ApplyIRToSubstitutions() local
238 auto &all_nodes = manager->all_nodes(); in ApplySubstitutionToIR() local
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/
Dacl_model_multi.cc50 std::vector<AnfNodePtr> all_nodes = TopoSort(fg->get_return()); in HasMultiGraph() local
/third_party/mindspore/mindspore/core/ir/
Dmanager.h314 AnfNodeSet &all_nodes() { return all_nodes_; } in all_nodes() function
Dfunc_graph.cc655 auto &all_nodes = nodes(); in EraseUnusedNodeInOrder() local
Dmanager.cc956 const AnfNodeSet &all_nodes = curr->nodes(); in RealRecompute() local
/third_party/mesa3d/src/gallium/drivers/r600/sb/
Dsb_shader.h277 std::vector<node*> all_nodes; variable
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc238 auto all_nodes = main_graph_->nodes(); in BroadCastColoring() local
819 std::vector<AnfNodePtr> all_nodes = DeepScopedGraphSearch(ret); in CutBorder() local
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Devaluator.cc164 …const auto &all_nodes = TopoSort(func_node, SuccIncoming, [](const AnfNodePtr &node) -> IncludeTyp… in LaunchRecursiveEval() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Dkprim.cc200 std::vector<AnfNodePtr> all_nodes = DeepScopedGraphSearch(fg->get_return()); in ReplacePythonOps() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/gnn/
Dgraph_data_impl.cc317 const std::vector<NodeIdType> &all_nodes = node_type_map_[neg_neighbor_type]; in GetNegSampledNeighbors() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dascend_kernel_compile.cc520 auto all_nodes = kernel_graph->execution_order(); in GetAllAscendNodes() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_auto_monad.cc1628 std::set<CNodePtr> all_nodes; in GetAllNodes() local
1678 std::set<CNodePtr> all_nodes = GetAllNodes(&search_list); in EraseParameter() local
Dsession_basic.cc365 auto all_nodes = DeepLinkedGraphSearch(ret); in ExistSummaryNode() local