Home
last modified time | relevance | path

Searched refs:FuncGraphSet (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore/core/ir/
Dmanager.h61 using FuncGraphSetPair = std::pair<FuncGraphPtr, FuncGraphSet>;
62 using FuncGraphSetPtr = std::shared_ptr<FuncGraphSet>;
81 using FuncGraphToFuncGraphSetMap = OrderedMap<FuncGraphPtr, FuncGraphSet>;
299 void MaybeDropFuncGraphs(const FuncGraphSet &func_graphs, bool ignore_users = false);
310 const FuncGraphSet &roots() const { return roots_; } in roots()
312 const FuncGraphSet &func_graphs() const { return func_graphs_; } in func_graphs()
322 FuncGraphSet &func_graph_parents_total(const FuncGraphPtr &fg) const;
324 FuncGraphSet &scopes(const FuncGraphPtr &fg) const;
328 FuncGraphSet &children(const FuncGraphPtr &fg) const;
330 FuncGraphSet &func_graphs_used_total(const FuncGraphPtr &fg) const;
[all …]
Dmanager.cc216 func_graphs_ = FuncGraphSet(); in Reset()
232 roots_ = FuncGraphSet(); in Init()
239 FuncGraphSet &FuncGraphManager::func_graph_parents_total(const FuncGraphPtr &fg) const { in func_graph_parents_total()
260 FuncGraphSet &FuncGraphManager::children(const FuncGraphPtr &fg) const { in children()
268 FuncGraphSet &FuncGraphManager::scopes(const FuncGraphPtr &fg) const { in scopes()
283 FuncGraphSet &FuncGraphManager::func_graphs_used_total(const FuncGraphPtr &fg) const { in func_graphs_used_total()
385 FuncGraphSet roots(func_graphs); in KeepRoots()
395 FuncGraphSet keep; in KeepRoots()
408 FuncGraphSet roots(func_graphs); in KeepRoots()
433 void FuncGraphManager::MaybeDropFuncGraphs(const FuncGraphSet &func_graphs, bool ignore_users) { in MaybeDropFuncGraphs()
[all …]
Dfunc_graph.h314 const FuncGraphSet &func_graphs_used_total();
327 const FuncGraphSet &children();
330 const FuncGraphSet &scope();
Dfunc_graph.cc380 const FuncGraphSet &FuncGraph::func_graphs_used_total() { in func_graphs_used_total()
470 const FuncGraphSet &FuncGraph::children() { in children()
476 const FuncGraphSet &FuncGraph::scope() { in scope()
735 FuncGraphSet graphs = graph_manager->func_graphs(); in ContainMultiTarget()
Dfunc_graph_cloner.h111 FuncGraphSet graph_set_;
Danf.h71 using FuncGraphSet = OrderedSet<FuncGraphPtr>; variable
/third_party/mindspore/tests/ut/cpp/parallel/
Dstep_parallel_test.cc368 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/vm/
Dtransform.cc420 const FuncGraphManagerPtr &manager_ptr, FuncGraphTransaction *const tr, const FuncGraphSet &fgs, in TraverseGraphMap()
550 FuncGraphSet graphs = prim_graph->manager()->func_graphs(); in CompileAndLink()
Dbackend.cc389 FuncGraphSet sub_graphs = root_graph->manager()->func_graphs(); in CompileGraphs()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dclean.cc669 FuncGraphSet all_graph = manager->func_graphs(); in EraseTuple()
/third_party/mindspore/tests/ut/cpp/ir/
Dmanager_test.cc578 FuncGraphSet s; in TEST_F()