Home
last modified time | relevance | path

Searched refs:IncludeFunc (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/core/ir/
Dgraph_utils.h40 using IncludeFunc = std::function<IncludeType(const AnfNodePtr &)>; variable
43 using SearchFunc = std::function<std::vector<AnfNodePtr>(const AnfNodePtr &, const IncludeFunc &)>;
46 std::vector<AnfNodePtr> DeepScopedGraphSearch(const AnfNodePtr &root, const IncludeFunc &include);
47 std::vector<AnfNodePtr> DeepUsedGraphSearch(const AnfNodePtr &root, const IncludeFunc &include);
48 std::vector<AnfNodePtr> DeepLinkedGraphSearch(const AnfNodePtr &root, const IncludeFunc &include);
60 std::vector<AnfNodePtr> DeepScopedGraphSearch(const AnfNodePtr &root, const IncludeFunc &include = …
61 std::vector<AnfNodePtr> DeepUsedGraphSearch(const AnfNodePtr &root, const IncludeFunc &include = Al…
62 std::vector<AnfNodePtr> DeepLinkedGraphSearch(const AnfNodePtr &root, const IncludeFunc &include = …
64 std::vector<AnfNodePtr> DeepScopedGraphSearchWithFilter(const AnfNodePtr &root, const IncludeFunc &…
69 std::vector<AnfNodePtr> DeepUsersSearch(const AnfNodePtr &root, const IncludeFunc &include,
[all …]
Dgraph_utils_extends.cc39 explicit DeepFirstSearcher(const IncludeFunc &include, const FilterFunc &filter = nullptr) in DeepFirstSearcher()
74 IncludeFunc include_;
81 explicit DeepScopedGraphSearcher(const IncludeFunc &include) : DeepFirstSearcher(include) {} in DeepScopedGraphSearcher()
126 explicit DeepUsedGraphSearcher(const IncludeFunc &include) : DeepFirstSearcher(include) {} in DeepUsedGraphSearcher()
151 explicit DeepLinkedGraphSearcher(const IncludeFunc &include) : DeepFirstSearcher(include) {} in DeepLinkedGraphSearcher()
166 explicit DeepUsersSearcher(const IncludeFunc &include, const FuncGraphManagerPtr &mng) in DeepUsersSearcher()
184 std::vector<AnfNodePtr> DeepScopedGraphSearch(const AnfNodePtr &root, const IncludeFunc &include) { in DeepScopedGraphSearch()
188 std::vector<AnfNodePtr> DeepScopedGraphSearchWithFilter(const AnfNodePtr &root, const IncludeFunc &… in DeepScopedGraphSearchWithFilter()
193 std::vector<AnfNodePtr> DeepUsedGraphSearch(const AnfNodePtr &root, const IncludeFunc &include) { in DeepUsedGraphSearch()
197 std::vector<AnfNodePtr> DeepLinkedGraphSearch(const AnfNodePtr &root, const IncludeFunc &include) { in DeepLinkedGraphSearch()
[all …]
Dgraph_utils.cc52 std::vector<AnfNodePtr> TopoSort(const AnfNodePtr &root, const SuccFunc &succ, const IncludeFunc &i… in TopoSort()
283 FuncGraphIndex::FuncGraphIndex(const FuncGraphPtr &fg, const SearchFunc &search, const IncludeFunc in FuncGraphIndex()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dupdate_state_formatter.cc195 auto IncludeFunc = [&result, &index_node](const AnfNodePtr &node) { in FindAllOutputs() local
202 static_cast<void>(DeepLinkedGraphSearch(out_node, IncludeFunc)); in FindAllOutputs()