Home
last modified time | relevance | path

Searched refs:func_graphs (Results 1 – 25 of 28) sorted by relevance

12

/third_party/mindspore/tests/ut/cpp/ir/
Dmanager_test.cc345 auto size = mng->func_graphs().size(); in CheckAnalysisSize()
386 ASSERT_EQ(2, mng->func_graphs().size()); in TEST_F()
426 ASSERT_EQ(3, mng->func_graphs().size()); in TEST_F()
490 ASSERT_EQ(3, mng->func_graphs().size()); in TEST_F()
576 const auto &fgs = mng->func_graphs(); in TEST_F()
589 ASSERT_EQ(mng->func_graphs().size(), (size_t)1); in TEST_F()
590 ASSERT_TRUE(mng->func_graphs().contains(fg1)); in TEST_F()
593 ASSERT_EQ(mng->func_graphs().size(), 2); in TEST_F()
594 ASSERT_TRUE(mng->func_graphs().contains(fg2)); in TEST_F()
597 ASSERT_EQ(mng->func_graphs().size(), 1); in TEST_F()
[all …]
/third_party/mindspore/mindspore/core/ir/
Dmanager.cc166 FuncGraphManagerPtr MakeManager(const std::vector<FuncGraphPtr> &func_graphs, bool manage) { in MakeManager() argument
167 auto m = std::make_shared<FuncGraphManager>(func_graphs, manage); in MakeManager()
172 FuncGraphManagerPtr Manage(const std::vector<FuncGraphPtr> &func_graphs, bool manage) { in Manage() argument
176 for (auto &fg : func_graphs) { in Manage()
192 for (auto &fg : func_graphs) { in Manage()
202 std::vector<FuncGraphPtr> func_graphs = {func_graph}; in Manage() local
203 return Manage(func_graphs, manage); in Manage()
370 void FuncGraphManager::KeepRoots(const std::vector<FuncGraphPtr> &func_graphs) { in KeepRoots() argument
373 for (auto &item : func_graphs) { in KeepRoots()
384 if (root_exist || func_graphs.empty()) { in KeepRoots()
[all …]
Dfunc_graph_cloner.cc33 Cloner::Cloner(const FuncGraphVector &func_graphs, bool clone_all_valuenodes, bool clone_all_child_… in Cloner() argument
40 for (auto &func_graph : func_graphs) { in Cloner()
513 const auto &func_graphs = BroadFirstSearchGraphUsed(lift_top_func_graph); in LiftParameters() local
514 for (auto &func_graph : func_graphs) { in LiftParameters()
517 Lift(func_graphs); in LiftParameters()
518 for (auto &func_graph : func_graphs) { in LiftParameters()
582 FuncGraphVector func_graphs; in Run() local
583 (void)std::transform(todo_.begin(), todo_.end(), std::back_inserter(func_graphs), in Run()
585 manager_ = Manage(func_graphs, false); in Run()
721 FuncGraphVector func_graphs = {func_graph}; in SpecializerClone() local
[all …]
Dmanager.h71 FuncGraphManagerPtr Manage(const std::vector<FuncGraphPtr> &func_graphs, bool manage = true);
73 FuncGraphManagerPtr MakeManager(const std::vector<FuncGraphPtr> &func_graphs = {}, bool manage = tr…
299 void MaybeDropFuncGraphs(const FuncGraphSet &func_graphs, bool ignore_users = false);
312 const FuncGraphSet &func_graphs() const { return func_graphs_; } in func_graphs() function
Dgraph_utils.cc298 std::set<FuncGraphPtr> func_graphs; in GetFuncGraphs() local
300 func_graphs = index_func_graph_[key]; in GetFuncGraphs()
302 return func_graphs; in GetFuncGraphs()
Dfunc_graph.cc330 std::vector<FuncGraphPtr> func_graphs; in free_variables_func_graphs() local
335 func_graphs.push_back(utils::cast<FuncGraphPtr>(key)); in free_variables_func_graphs()
339 return func_graphs; in free_variables_func_graphs()
735 FuncGraphSet graphs = graph_manager->func_graphs(); in ContainMultiTarget()
Dfunc_graph_cloner.h47 explicit Cloner(const FuncGraphVector &func_graphs = {}, bool clone_all_valuenodes = false,
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dcontrol_node_parser.cc320 std::vector<FuncGraphPtr> func_graphs = FetchFuncGraphbyCallNode(real_output.first); in FetchFuncGraphOutput() local
321 for (const auto &graph : func_graphs) { in FetchFuncGraphOutput()
339 const auto func_graphs = FetchFuncGraphbyCallNode(call_node); in FetchOutputByCallNode() local
341 for (const auto func_graph : func_graphs) { in FetchOutputByCallNode()
495 const auto &func_graphs = FetchFuncGraphbyCallNode(real_node); in FetchRealOutputByCallNode() local
496 for (const auto &func_graph : func_graphs) { in FetchRealOutputByCallNode()
577 std::vector<FuncGraphPtr> func_graphs; in FetchFuncGraphbyCallNode() local
579 return func_graphs; in FetchFuncGraphbyCallNode()
589 (void)func_graphs.emplace_back(GetFuncGraphFromPartial(cnode_inputs[i])); in FetchFuncGraphbyCallNode()
591 (void)func_graphs.emplace_back(GetValueNode<FuncGraphPtr>(cnode_inputs[i])); in FetchFuncGraphbyCallNode()
[all …]
Dgraph_scheduler.cc1781 std::vector<FuncGraphPtr> func_graphs = FetchFuncGraphbyCallNode(cnode); in LinkDataArrowByCallInput() local
1784 for (const auto &func_graph : func_graphs) { in LinkDataArrowByCallInput()
2078 const auto &func_graphs = FetchFuncGraphbyCallNode(call_node); in LinkControlArrowForSwitchActor() local
2079 for (const auto func_graph : func_graphs) { in LinkControlArrowForSwitchActor()
/third_party/mindspore/tests/ut/cpp/pipeline/parse/
Dresolve_test.cc54 ASSERT_EQ(manager->func_graphs().size(), (size_t)2); in TEST_F()
70 ASSERT_EQ(manager->func_graphs().size(), (size_t)2); in TEST_F()
/third_party/mindspore/tests/ut/cpp/parallel/
Dstep_parallel_test.cc200 std::vector<FuncGraphPtr> func_graphs{func_graph}; in Make_Manager() local
201 FuncGraphManagerPtr manager = std::make_shared<FuncGraphManager>(func_graphs, true); in Make_Manager()
368 FuncGraphSet graphs = manager->func_graphs(); in TEST_F()
377 FuncGraphSet graphs = manager->func_graphs(); in TEST_F()
386 FuncGraphSet graphs = manager->func_graphs(); in TEST_F()
405 FuncGraphSet graphs = manager->func_graphs(); in TEST_F()
444 FuncGraphSet graphs = manager->func_graphs(); in TEST_F()
467 FuncGraphSet graphs = manager->func_graphs(); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Doptimizer.cc105 std::vector<FuncGraphPtr> func_graphs; in Optimize() local
106 func_graphs.push_back(func_graph); in Optimize()
/third_party/mindspore/tests/ut/cpp/transform/
Dconvert_test.cc130 auto anfGraph = *(manager->func_graphs().begin()); in TEST_F()
194 auto anf_graph = *(manager->func_graphs().begin()); in TEST_F()
222 auto anf_graph = *(manager->func_graphs().begin()); in TEST_F()
245 auto anf_graph = *(manager->func_graphs().begin()); in TEST_F()
267 auto anf_graph = *(manager->func_graphs().begin()); in TEST_F()
289 auto anf_graph = *(manager->func_graphs().begin()); in TEST_F()
317 auto anf_graph = *(manager->func_graphs().begin()); in TEST_F()
617 auto anfGraph = *(manager->func_graphs().begin()); in TEST_F()
751 auto anfGraph = *(manager->func_graphs().begin()); in TEST_F()
Dtransform_base_test.cc45 for (auto func_graph : manager->func_graphs()) { in getAnfGraph()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dpy_pass.cc290 auto func_graphs = manager->func_graphs(); in Run() local
292 for (auto &fg : func_graphs) { in Run()
Dauto_monad_eliminate.cc283 for (const FuncGraphPtr &fg : manager->func_graphs()) { in ReplaceAutoMonadNode()
327 for (const FuncGraphPtr &fg : manager->func_graphs()) { in EliminateAutoMonadNode()
Dcse.cc116 for (FuncGraphPtr fg : manager->func_graphs()) { in BuildOrderGroupAndDoReplace()
Dclean.cc669 FuncGraphSet all_graph = manager->func_graphs(); in EraseTuple()
/third_party/mindspore/tests/ut/cpp/optimizer/
Dcconv_test.cc29 for (auto &iter : mng->func_graphs()) { in CheckNoFreeVariables()
/third_party/mindspore/mindspore/lite/tools/converter/import/
Dprimitive_adjust.cc543 bool PrimitiveAdjust::Run(const FuncGraphPtr &func_graphs) { in Run() argument
544 MS_ASSERT(func_graphs != nullptr); in Run()
550 static auto root_func_manager = Manage(func_graphs); in Run()
552 lite::GetAllFuncGraph(func_graphs, &all_func_graphs); in Run()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Daction.cc94 size_t graph_nums = manager->func_graphs().size(); in ResetMindRTEnable()
272 auto all_graphs = manager->func_graphs(); in GetLoadedGraph()
392 FuncGraphVector func_graphs = {fg}; in CombineLikeGraphs() local
393 …ClonerPtr cloner = std::make_shared<Cloner>(func_graphs, false, false, true, std::make_shared<Trac… in CombineLikeGraphs()
717 auto graphs = manager->func_graphs(); in TaskEmitAction()
Dpass.cc617 for (auto &fg : manager->func_graphs()) { in RemoveValueNodeDuplicationsPass()
/third_party/mindspore/mindspore/ccsrc/vm/
Dtransform.cc491 auto &fgs = manager_ptr->func_graphs(); in WrapPrimitives()
550 FuncGraphSet graphs = prim_graph->manager()->func_graphs(); in CompileAndLink()
Dvmimpl.cc125 auto graphs = graph->manager()->func_graphs(); in AcquireGraph()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc63 for (auto &fg : manager_->func_graphs()) { in MainGraph()
199 for (auto &fg : manager_->func_graphs()) { in Coloring()
920 for (auto &fg : manager_->func_graphs()) { in ElimGraphStage()

12