Home
last modified time | relevance | path

Searched refs:FuncGraphManagerPtr (Results 1 – 25 of 112) sorted by relevance

12345

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.h31 void InsertVirtualAssignAdd(const std::pair<AnfNodePtr, int> &node_user, const FuncGraphManagerPtr
33 void InsertVirtualAccuGrad(const AnfNodePtr &recv, const FuncGraphManagerPtr &manager, const AnfNod…
48 …pend(const AnfNodePtr &prior_node, const AnfNodePtr &post_node, const FuncGraphManagerPtr &manager,
57 void ReorderForPredict(const FuncGraphPtr &root, const FuncGraphManagerPtr &manager);
58 void HandleMicroBatch(const std::vector<AnfNodePtr> &all_nodes, const FuncGraphManagerPtr &manager);
60 void LabelNeedGrad(const FuncGraphManagerPtr &manager, const FuncGraphPtr &root);
63 void LastStageEndNode(const std::vector<AnfNodePtr> &all_nodes, const FuncGraphManagerPtr &manager,
66 void ParameterStartNode(const std::vector<AnfNodePtr> &all_nodes, const FuncGraphManagerPtr &manage…
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dredundant_op_remove_pass.h33 int ReplaceOp(const AnfNodePtr &anf_node, const FuncGraphManagerPtr &manager);
35 int ReplaceTupleGetItem(const AnfNodePtr &anf_node, const FuncGraphManagerPtr &manager);
36 int RemoveDropoutOp(const AnfNodePtr &anf_node, const FuncGraphManagerPtr &manager);
37 int RemoveInvalidPadOp(const AnfNodePtr &anf_node, const FuncGraphManagerPtr &manager);
38 int RemoveInvalidTransposeOp(const AnfNodePtr &anf_node, const FuncGraphManagerPtr &manager);
Dredundant_op_remove_pass.cc136 int RemoveRedundantOpPass::ReplaceOp(const AnfNodePtr &anf_node, const FuncGraphManagerPtr &manager… in ReplaceOp()
189 …undantOpPass::ReplaceTupleGetItem(const AnfNodePtr &anf_node, const FuncGraphManagerPtr &manager) { in ReplaceTupleGetItem()
227 int RemoveRedundantOpPass::RemoveDropoutOp(const AnfNodePtr &anf_node, const FuncGraphManagerPtr &m… in RemoveDropoutOp()
288 int RemoveRedundantOpPass::RemoveInvalidPadOp(const AnfNodePtr &anf_node, const FuncGraphManagerPtr in RemoveInvalidPadOp()
342 …tOpPass::RemoveInvalidTransposeOp(const AnfNodePtr &anf_node, const FuncGraphManagerPtr &manager) { in RemoveInvalidTransposeOp()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dcse.h41 virtual bool Cse(const FuncGraphPtr root, const FuncGraphManagerPtr manager) const;
44 …bool BuildOrderGroupAndDoReplaceForOneGraph(const FuncGraphPtr &fg, const FuncGraphManagerPtr &man…
47 bool BuildOrderGroupAndDoReplace(const FuncGraphManagerPtr manager) const;
48 bool DoReplace(const FuncGraphManagerPtr manager, const std::vector<std::size_t> &order_group,
Dclean.h34 bool SimplifyDataStructures(const FuncGraphPtr &root, const FuncGraphManagerPtr &manager);
35 bool CleanAfterOptA(const FuncGraphPtr &root, const FuncGraphManagerPtr &manager);
39 void EraseTuple(const FuncGraphPtr &root, const FuncGraphManagerPtr &manager);
Dauto_monad_eliminate.cc126 void DeleteLoadUserUpdateState(const FuncGraphManagerPtr &manager, const AnfNodePtr &load_user) { in DeleteLoadUserUpdateState()
144 void DeleteLoadUserMakeTuple(const FuncGraphManagerPtr &manager, const CNodePtr &make_tuple, const … in DeleteLoadUserMakeTuple()
169 void ReplaceLoadUserMakeTuple(const FuncGraphManagerPtr &manager, const FuncGraphPtr &fg, const CNo… in ReplaceLoadUserMakeTuple()
185 bool ReplaceLoadUser(const FuncGraphManagerPtr &manager, const FuncGraphPtr &fg, const AnfNodePtr &… in ReplaceLoadUser()
220 bool ReplaceSameGroupLoad(const FuncGraphManagerPtr &manager, const FuncGraphPtr &fg, in ReplaceSameGroupLoad()
281 bool AutoMonadEliminator::ReplaceAutoMonadNode(const FuncGraphManagerPtr &manager) const { in ReplaceAutoMonadNode()
325 bool AutoMonadEliminator::EliminateAutoMonadNode(const FuncGraphManagerPtr &manager) const { in EliminateAutoMonadNode()
Dauto_monad_eliminate.h43 bool ReplaceAutoMonadNode(const FuncGraphManagerPtr &manager) const;
44 bool EliminateAutoMonadNode(const FuncGraphManagerPtr &manager) const;
Dcse.cc70 bool CSE::BuildOrderGroupAndDoReplaceForOneGraph(const FuncGraphPtr &fg, const FuncGraphManagerPtr in BuildOrderGroupAndDoReplaceForOneGraph()
114 bool CSE::BuildOrderGroupAndDoReplace(const FuncGraphManagerPtr manager) const { in BuildOrderGroupAndDoReplace()
213 bool CSE::DoReplace(const FuncGraphManagerPtr manager, const std::vector<std::size_t> &order_group, in DoReplace()
261 bool CSE::Cse(const FuncGraphPtr root, const FuncGraphManagerPtr manager) const { in Cse()
Dgraph_transform.h42 FuncGraphPtr operator()(const FuncGraphPtr &fg, const FuncGraphManagerPtr &mng) { in operator()
71 FuncGraphPtr TransformGraphParam(const FuncGraphPtr &fg, const FuncGraphManagerPtr &mng) { in TransformGraphParam()
Drecompute.cc81 std::vector<CNodePtr> FindCandidateRecomputedNodes(const FuncGraphManagerPtr &mng, in FindCandidateRecomputedNodes()
111 void GetMaxSubGraph(const FuncGraphManagerPtr &mng, std::unordered_set<CNodePtr> *recomputed_nodes,… in GetMaxSubGraph()
162 void GetOriginRecomputeAndTargetNodes(const FuncGraphManagerPtr &mng, in GetOriginRecomputeAndTargetNodes()
241 bool HasForwardOutput(const FuncGraphManagerPtr &mng, const AnfNodePtr &node) { in HasForwardOutput()
253 void GetTupleGetItemOutputNodes(const FuncGraphManagerPtr &mng, const AnfNodePtr &node, in GetTupleGetItemOutputNodes()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dresource_base.h37 FuncGraphManagerPtr manager() { return manager_; } in manager()
39 void set_manager(const FuncGraphManagerPtr &manager) { manager_ = manager; } in set_manager()
55 FuncGraphManagerPtr manager_;
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dadd_atomic_clean.h84 …lean(const KernelGraphPtr &main_graph, const AnfNodePtr &anf_node, const FuncGraphManagerPtr &mng);
90 const FuncGraphManagerPtr &mng);
99 const FuncGraphManagerPtr &mng,
102 const FuncGraphManagerPtr &mng);
Dreorder_ops.h47 …bool ReorderTypeInsensitiveCastDown(const FuncGraphPtr &func_graph, const FuncGraphManagerPtr &mng,
49 bool ReorderCastUpTypeInsensitive(const FuncGraphPtr &func_graph, const FuncGraphManagerPtr &mng,
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_context.h40 void AddManager(const FuncGraphManagerPtr &m) { manager_list_.push_back(m); } in AddManager()
43 std::vector<FuncGraphManagerPtr> manager_list_;
/third_party/mindspore/mindspore/core/api/ir/
Dfunc_graph_manager.h35 using FuncGraphManagerPtr = std::shared_ptr<FuncGraphManager>; variable
50 static FuncGraphManagerPtr Manage(const FuncGraphPtr &func_graph, bool manage = true);
/third_party/mindspore/mindspore/core/ir/
Dgraph_utils_extends.cc166 explicit DeepUsersSearcher(const IncludeFunc &include, const FuncGraphManagerPtr &mng) in DeepUsersSearcher()
179 FuncGraphManagerPtr mng_;
202 const FuncGraphManagerPtr &mng) { in DeepUsersSearch()
Dmanager.h56 using FuncGraphManagerPtr = std::shared_ptr<FuncGraphManager>; variable
69 FuncGraphManagerPtr Manage(FuncGraphPtr func_graph, bool manage = true);
71 FuncGraphManagerPtr Manage(const std::vector<FuncGraphPtr> &func_graphs, bool manage = true);
73 FuncGraphManagerPtr MakeManager(const std::vector<FuncGraphPtr> &func_graphs = {}, bool manage = tr…
Dgraph_utils.h68 using FuncGraphManagerPtr = std::shared_ptr<FuncGraphManager>; variable
70 const FuncGraphManagerPtr &mng);
/third_party/mindspore/mindspore/lite/tools/converter/acl/
Dacl_pass.h44 STATUS DeparseGraph(const FuncGraphPtr &func_graph, const FuncGraphManagerPtr &manager);
52 STATUS ModifyGraphByCustomNode(const FuncGraphPtr &func_graph, const FuncGraphManagerPtr &manager,
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/
Dresolve.cc243 AnfNodePtr TransformToMakeTupleNodes(const FuncGraphManagerPtr &manager, const FuncGraphPtr &func_g… in TransformToMakeTupleNodes()
269 bool TransformVectorFuncValueNode(const FuncGraphManagerPtr &manager, const FuncGraphPtr &func_grap… in TransformVectorFuncValueNode()
291 AnfNodePtr ResolveObjectAndAddToManager(const FuncGraphManagerPtr &manager, const py::object &obj, in ResolveObjectAndAddToManager()
313 AnfNodePtr ResolveSymbol(const FuncGraphManagerPtr &manager, const NameSpacePtr &name_space, const … in ResolveSymbol()
327 AnfNodePtr ResolveCellwithAttr(const FuncGraphManagerPtr &manager, const NameSpacePtr &name_space, in ResolveCellwithAttr()
391 bool ResolveAll(const FuncGraphManagerPtr &manager) { in ResolveAll()
Dresolve.h177 AnfNodePtr ResolveSymbol(const FuncGraphManagerPtr &manager, const NameSpacePtr &name_space, const …
181 AnfNodePtr ResolveCellwithAttr(const FuncGraphManagerPtr &manager, const NameSpacePtr &name_space,
189 bool ResolveAll(const FuncGraphManagerPtr &manager);
/third_party/mindspore/tests/ut/cpp/parallel/
Dstep_parallel_test.cc117 FuncGraphManagerPtr Make_Manager(int64_t condition = 0) { in Make_Manager()
201 FuncGraphManagerPtr manager = std::make_shared<FuncGraphManager>(func_graphs, true); in Make_Manager()
367 FuncGraphManagerPtr manager = Make_Manager(); in TEST_F()
376 FuncGraphManagerPtr manager = Make_Manager(2); in TEST_F()
385 FuncGraphManagerPtr manager = Make_Manager(3); in TEST_F()
404 FuncGraphManagerPtr manager = Make_Manager(); in TEST_F()
443 FuncGraphManagerPtr manager = Make_Manager(); in TEST_F()
466 FuncGraphManagerPtr manager = Make_Manager(); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.h43 …PipelineTransformer(const FuncGraphManagerPtr &manager, int stage, const FuncGraphPtr &root, int64… in PipelineTransformer()
88 FuncGraphManagerPtr manager_;
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dparameter_eliminate.h102 …red_set<size_t> EraseUnusedParameters(const FuncGraphPtr &fg, const FuncGraphManagerPtr &manager) { in EraseUnusedParameters()
130 const FuncGraphManagerPtr &manager) { in EraseArgs()
/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Dcommon_subexpression_elimination_test.cc63 FuncGraphManagerPtr manager = Manage(func_graph); in TEST_F()
91 FuncGraphManagerPtr manager = Manage(func_graph); in TEST_F()
134 FuncGraphManagerPtr manager = Manage(func_graph); in TEST_F()

12345