p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!22374 Add 'Create' method for api::FuncGraph 

Merge pull request !22374 from hewei/core_user_api
This commit is contained in:
i-robot 2021-08-26 06:42:15 +00:00 committed by Gitee
parent 8e39dd4ec7 c1f124cecd
commit 3d10610bc9
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/core/api/ir/func_graph.h
View File

@ -48,6 +48,8 @@ class MS_CORE_API FuncGraph {
virtual FuncGraphManagerPtr get_manager() const = 0;
static std::vector<AnfNodePtr> TopoSort(const AnfNodePtr &node);
static FuncGraphPtr Create();
};
} // namespace mindspore::api
#endif // MINDSPORE_CORE_API_IR_FUNC_GRAPH_H_

3
mindspore/core/ir/func_graph.cc
View File

@ -743,5 +743,8 @@ size_t NewFgSeenGeneration() {
// Implement TopoSort api.
std::vector<AnfNodePtr> api::FuncGraph::TopoSort(const AnfNodePtr &node) { return mindspore::TopoSort(node); }
// Create an api::FuncGraph instance.
api::FuncGraphPtr api::FuncGraph::Create() { return std::make_shared<mindspore::FuncGraph>(); }
const PrimitivePtr FuncGraphTransform::func_graph_prim_ = std::make_shared<Primitive>("FuncGraph");
} // namespace mindspore