Searched refs:IncludeFunc (Results 1 – 4 of 4) sorted by relevance
/third_party/mindspore/mindspore/core/ir/ |
D | graph_utils.h | 40 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 …]
|
D | graph_utils_extends.cc | 39 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 …]
|
D | graph_utils.cc | 52 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/ |
D | update_state_formatter.cc | 195 auto IncludeFunc = [&result, &index_node](const AnfNodePtr &node) { in FindAllOutputs() local 202 static_cast<void>(DeepLinkedGraphSearch(out_node, IncludeFunc)); in FindAllOutputs()
|