Home
last modified time | relevance | path

Searched refs:FuncGraphPtr (Results 1 – 25 of 911) sorted by relevance

12345678910>>...37

/third_party/mindspore/mindspore/core/ir/
Dfunc_graph_cloner.h40 FuncGraphPtr origin;
41 FuncGraphPtr target;
52 void AddClone(const FuncGraphPtr &func_graph, const FuncGraphPtr &target_func_graph = nullptr,
59 FuncGraphPtr operator[](const FuncGraphPtr &func_graph);
63 std::unordered_map<FuncGraphPtr, FuncGraphPtr> &cloned_func_graph() { return repl_func_graph_; } in cloned_func_graph()
70 std::unordered_map<FuncGraphPtr, FuncGraphPtr> repl_func_graph_;
76 void CloneNode(const AnfNodePtr &node, const FuncGraphPtr &target);
78 void CloneValueNode(const AnfNodePtr &node, const FuncGraphPtr &target);
79 void CloneCNode(const AnfNodePtr &node, const FuncGraphPtr &target);
80 void CloneParameter(const AnfNodePtr &node, const FuncGraphPtr &target, bool is_add = false);
[all …]
Dmanager.h61 using FuncGraphSetPair = std::pair<FuncGraphPtr, FuncGraphSet>;
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…
81 using FuncGraphToFuncGraphSetMap = OrderedMap<FuncGraphPtr, FuncGraphSet>;
101 void Recompute(const FuncGraphPtr &fg);
105 bool IsValidate(const FuncGraphPtr &fg) { return func_graphs_validate_[fg]; } in IsValidate()
112 virtual void RealRecompute(FuncGraphPtr) {} in RealRecompute() argument
116 OrderedMap<FuncGraphPtr, bool> func_graphs_validate_;
137 void RealRecompute(FuncGraphPtr fg) override;
[all …]
/third_party/mindspore/tests/ut/cpp/optimizer/
Dlib_test.cc47 FuncGraphPtr RunTransform(FuncGraphPtr gbefore, const SubstitutionList &transform) { in RunTransform()
51 FuncGraphPtr gbefore_clone = BasicClone(gbefore); in RunTransform()
56 FuncGraphPtr RunSubs(FuncGraphPtr before, std::vector<SubstitutionPtr> opts = {}) { in RunSubs()
60 bool CheckTransform(FuncGraphPtr gbefore, FuncGraphPtr gafter, const SubstitutionList &transform, in CheckTransform()
65 FuncGraphPtr gbefore_clone = BasicClone(gbefore); in CheckTransform()
70 bool CheckOpt(FuncGraphPtr before, FuncGraphPtr after, std::vector<SubstitutionPtr> opts = {}, in CheckOpt()
87 FuncGraphPtr before1 = getPyFun.CallAndParseRet("test_simplify_always_true_false", "before_1"); in TEST_F()
88 FuncGraphPtr before2 = getPyFun.CallAndParseRet("test_simplify_always_true_false", "before_2"); in TEST_F()
89 FuncGraphPtr after = getPyFun.CallAndParseRet("test_simplify_always_true_false", "after"); in TEST_F()
96 FuncGraphPtr before1 = getPyFun.CallAndParseRet("test_inline", "before"); in TEST_F()
[all …]
Dcconv_test.cc27 void CheckNoFreeVariables(FuncGraphPtr root) { in CheckNoFreeVariables()
49 void CheckCconv(FuncGraphPtr g) { in CheckCconv()
72 FuncGraphPtr func_graph = getPyFun.CallAndParseRet("get_test_cconv_fn", "test_straight"); in TEST_F()
78 FuncGraphPtr func_graph = getPyFun.CallAndParseRet("get_test_cconv_fn", "test_simple_closure"); in TEST_F()
84 FuncGraphPtr func_graph = getPyFun.CallAndParseRet("get_test_cconv_fn", "test_max"); in TEST_F()
90 FuncGraphPtr func_graph = getPyFun.CallAndParseRet("get_test_cconv_fn", "test_deep_nesting"); in TEST_F()
96 …FuncGraphPtr func_graph = getPyFun.CallAndParseRet("get_test_cconv_fn", "test_return_in_double_whi… in TEST_F()
102 FuncGraphPtr func_graph = getPyFun.CallAndParseRet("get_test_cconv_fn", "test_pow10"); in TEST_F()
108 …FuncGraphPtr func_graph = getPyFun.CallAndParseRet("get_test_cconv_fn", "test_closure_as_simple_fv… in TEST_F()
114 FuncGraphPtr func_graph = getPyFun.CallAndParseRet("get_test_cconv_fn", "test_closure_as_fv"); in TEST_F()
[all …]
Dopt_test.cc87 …bool CheckTransform(FuncGraphPtr gbefore, FuncGraphPtr gafter, const SubstitutionList &transform) { in CheckTransform()
91 FuncGraphPtr gbefore_clone = BasicClone(gbefore); in CheckTransform()
98 bool CheckOpt(FuncGraphPtr before, FuncGraphPtr after, std::vector<SubstitutionPtr> opts = {}) { in CheckOpt()
124 FuncGraphPtr before = getPyFun.CallAndParseRet("test_add_zero", "before_1"); in TEST_F()
132 FuncGraphPtr before = getPyFun.CallAndParseRet("test_add_zero", "before_1"); in TEST_F()
133 FuncGraphPtr after = getPyFun.CallAndParseRet("test_add_zero", "after"); in TEST_F()
141 FuncGraphPtr before = getPyFun.CallAndParseRet("test_add_zero", "before_2"); in TEST_F()
142 FuncGraphPtr after = getPyFun.CallAndParseRet("test_add_zero", "after"); in TEST_F()
150 FuncGraphPtr before = getPyFun.CallAndParseRet("test_elimR", "before_1"); in TEST_F()
151 FuncGraphPtr after = getPyFun.CallAndParseRet("test_elimR", "after"); in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Ddfunctor.h41 using Registry = std::unordered_map<PrimitivePtr, FuncGraphPtr, PrimitiveHasher, PrimitiveTotalEqua…
53 DFunctor(const FuncGraphPtr &primal_graph, const pipeline::ResourceBasePtr &resources);
59 FuncGraphPtr k_graph();
60 FuncGraphPtr tape();
62 FuncGraphPtr KUserDefined(const FuncGraphPtr &primal);
106 …void GetForwardOutNodeAndBpropGraph(const CNodePtr &k_app, CNodePtr *forward_node, FuncGraphPtr *b…
107 FuncGraphPtr *fprop_graph);
108 …std::vector<AnfNodePtr> RunOutputReplace(const CNodePtr &forward_node, const FuncGraphPtr &bprop_g…
109 … const FuncGraphPtr &fprop_graph, const CNodePtr &cnode_morph);
110 …std::vector<AnfNodePtr> RunInputReplace(const FuncGraphPtr &bprop_graph, const FuncGraphPtr &fprop…
[all …]
Dprim_bprop_optimizer.h54 std::unordered_map<TupleListKey, FuncGraphPtr, PrimitiveTupleListHasher, PrimitiveTupleListEqual>;
88 FuncGraphPtr opt_func_graph_;
102 …explicit PrimBpropOptGraphLevel2Info(const FuncGraphPtr &func_graph) : opt_func_graph_(func_graph)… in PrimBpropOptGraphLevel2Info()
105 const FuncGraphPtr &opt_func_graph() const { return opt_func_graph_; } in opt_func_graph()
124 FuncGraphPtr opt_func_graph_;
143 …FuncGraphPtr OptimizeBPropFuncGraph(const FuncGraphPtr &bprop_fg, const CNodePtr &c_node, const Va…
147 FuncGraphPtr BpropGraphFinalOpt(const pipeline::ResourcePtr &res) const;
163 PrimBpropOptGraphInfoPtr PrimBpropOptStep1(const FuncGraphPtr &bprop_fg);
166 PrimBpropOptGraphLevel2InfoPtr PrimBpropOptStep2(const FuncGraphPtr &bprop_fg,
169 …void BindAbsToParameters(const FuncGraphPtr &bprop_fg, const abstract::AbstractBasePtrList &abs_li…
[all …]
/third_party/mindspore/mindspore/lite/tools/converter/
Danf_transform.h37 FuncGraphPtr Transform(const FuncGraphPtr &old_graph, const converter::Flags *config = nullptr);
42 …FuncGraphPtr TransformFuncGraph(const FuncGraphPtr &old_graph, const converter::Flags *config = nu…
44 static int RunFusionPass(const FuncGraphPtr &old_graph, const converter::Flags *config);
46 static int RunGraphPass(const FuncGraphPtr &old_graph, const converter::Flags *config);
48 static int RunConvertPass(const FuncGraphPtr &old_graph, const converter::Flags *config);
50 static int RunConstFoldPass(const FuncGraphPtr &olde_graph, const converter::Flags *config);
52 static int RunParallelPass(const FuncGraphPtr &old_graph, const converter::Flags *config);
54 int DoQuantize(const FuncGraphPtr &old_graph, const converter::Flags *config);
56 …static void GetFuncGraphs(const FuncGraphPtr &func_graph, std::set<FuncGraphPtr> *all_func_graphs);
58 int DoSingleGraphQuantize(const FuncGraphPtr &old_graph, const converter::Flags *config);
[all …]
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dcontrol_flow_pass.h33 bool Run(const FuncGraphPtr &fg) override;
36 …void ReplaceNode(const FuncGraphPtr &fg, const std::unordered_map<AnfNodePtr, AnfNodePtr> &replace…
40 …int SplitGraph(const FuncGraphPtr &fg, AnfNodePtr *control_flow_node, std::set<AnfNodePtr> *visite…
46 int CreateAfterGraph(const FuncGraphPtr &main_fg, const std::vector<AnfNodePtr> &remain_nodes,
47 const CNodePtr &aim_cnode, FuncGraphPtr *after_fg);
51 …const FuncGraphPtr &fg, const CNodePtr &while_cnode, const std::vector<AnfNodePtr> &visited_nodes_…
54 …int CreateWhileBodyPartialNode(const FuncGraphPtr &cond_fg, const CNodePtr &while_cnode, CNodePtr …
56 …const FuncGraphPtr &main_fg, const FuncGraphPtr &cond_fg, const std::vector<AnfNodePtr> &remain_no…
60 int ProcessWhileOp(const FuncGraphPtr &fg, const std::set<AnfNodePtr> &visited_nodes,
64 int CreateIfPartialNodeExternalInputs(const CNodePtr &if_cnode, const FuncGraphPtr &partial_fg,
[all …]
Ddecrease_transpose_algo.h40 bool Run(const FuncGraphPtr &func_graph) override;
43 …STATUS InsertPostTransNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode, const std::vecto…
44 …STATUS InsertPreTransNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode, const std::vector…
45 …STATUS GenNewInput(const FuncGraphPtr &func_graph, const CNodePtr &cnode, const std::vector<int> p…
47 bool RunDoFixFormat(const FuncGraphPtr &func_graph, const CNodePtr &cnode);
48 bool DoFixFormat(const FuncGraphPtr &func_graph);
49 bool DecreaseTransposeForSingleOp(const FuncGraphPtr &func_graph);
50 bool DecreaseTransposeForMultiOp(const FuncGraphPtr &func_graph);
51 STATUS PostTransposeFusion(const FuncGraphPtr &func_graph, const CNodePtr &cnode);
53 STATUS HandleGraphMultiNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode,
[all …]
Dslice_prepose_pass.h39 bool Run(const FuncGraphPtr &graph) override;
44 static STATUS SwapSliceWithPreceed(const FuncGraphPtr &graph, const CNodePtr &slice_cnode,
48 static CNodePtr InsertSlice(const FuncGraphPtr &graph, const std::vector<AnfNodePtr> &inputs,
53 static CNodePtr CreateReshapeCNode(const FuncGraphPtr &graph, const std::vector<int64_t> &shape,
61 static bool PreposeWithNormalReshape(const FuncGraphPtr &graph, const CNodePtr &slice_cnode,
65 …static CNodePtr CreateSlice1ForReshapePrepose(const FuncGraphPtr &graph, const CNodePtr &slice_cno…
69 …static CNodePtr CreateSlice2ForReshapePrepose(const FuncGraphPtr &graph, const CNodePtr &slice_cno…
73 static bool PreposeWithAbnormalReshape(const FuncGraphPtr &graph, const CNodePtr &slice_cnode,
80 …static bool DoPrepose(const FuncGraphPtr &graph, const CNodePtr &slice_cnode, const CNodePtr &prec…
82 …static bool PreposeWithSoftmax(const FuncGraphPtr &graph, const CNodePtr &slice_cnode, const CNode…
[all …]
/third_party/mindspore/mindspore/lite/tools/converter/import/
Dmindir_control_flow_adjust.h34 bool Run(const FuncGraphPtr &graph);
37 std::vector<AnfNodePtr> GetFgOutput(const FuncGraphPtr &fg);
38 int ModifyFgToCallAfterFg(const FuncGraphPtr &fg, const FuncGraphPtr &after_fg);
39 bool HasCallAfter(const FuncGraphPtr &partial_fg);
40 …FuncGraphPtr AddAfterFuncGraph(const FuncGraphPtr &fg, const std::vector<AnfNodePtr> &one_of_inlin…
42 …int AddAfterFgForInlinedFg(const std::set<FuncGraphPtr> &all_func_graphs, const FuncGraphPtr &main…
43 int InsertPartialFusionForRawCall(const std::set<FuncGraphPtr> &all_func_graphs);
44 CNodePtr GetMainFgSwitchNode(const FuncGraphPtr &fg);
45 int ResetFuncGraph(const FuncGraphPtr &fg, std::set<FuncGraphPtr> all_func_graphs);
/third_party/mindspore/tests/ut/cpp/optimizer/ad/
Dad_test.cc44 FuncGraphPtr g = getPyFun(testCase); in AssertExpect()
47 FuncGraphPtr dg = Grad(g, resourcePtr); in AssertExpect()
51 …void AssertExpect(const std::string& testCase, const FuncGraphPtr& dg) { ASSERT_TRUE(dg != nullptr… in AssertExpect()
95 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarAdd), resourcePtr); in TEST_F()
100 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarMul), resourcePtr); in TEST_F()
105 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarSub), resourcePtr); in TEST_F()
110 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarDiv), resourcePtr); in TEST_F()
115 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarPow), resourcePtr); in TEST_F()
120 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarExp), resourcePtr); in TEST_F()
125 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarUadd), resourcePtr); in TEST_F()
[all …]
/third_party/mindspore/tests/ut/cpp/pipeline/parse/
Dparser_test.cc56 FuncGraphPtr func_graph = ParsePythonCode(fn_null); in TEST_F()
95 FuncGraphPtr func_graph = ParsePythonCode(fn); in TEST_F()
102 FuncGraphPtr ret_val = ParsePythonCode(fn); in TEST_F()
109 FuncGraphPtr ret_val = ParsePythonCode(fn); in TEST_F()
116 FuncGraphPtr ret_val = ParsePythonCode(fn); in TEST_F()
123 FuncGraphPtr ret_val = ParsePythonCode(fn); in TEST_F()
128 FuncGraphPtr ret_val; in TEST_F()
135 FuncGraphPtr ret_val; in TEST_F()
142 FuncGraphPtr ret_val; in TEST_F()
157 FuncGraphPtr func_graph = ParsePythonCode(fn); in TEST_F()
[all …]
/third_party/mindspore/mindspore/lite/tools/converter/acl/
Dacl_pass.h39 bool Run(const FuncGraphPtr &func_graph) override;
42 STATUS PreProcGraph(const FuncGraphPtr &func_graph);
43 STATUS PostProcGraph(const FuncGraphPtr &func_graph);
44 STATUS DeparseGraph(const FuncGraphPtr &func_graph, const FuncGraphManagerPtr &manager);
45 STATUS RunPrimitiveMapper(const FuncGraphPtr &func_graph);
46 STATUS BuildGraph(const FuncGraphPtr &func_graph);
47 STATUS ConvertGraphToOm(const FuncGraphPtr &func_graph, Buffer *om_data);
48 ParameterPtr CreateOmParameter(const FuncGraphPtr &func_graph, const Buffer &om);
49 CNodePtr CreateCustomNode(const FuncGraphPtr &func_graph);
50 STATUS SetCustomOutputs(const FuncGraphPtr &func_graph, const CNodePtr &custom_node);
[all …]
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/
Dadam_apply_one_fusion_test.cc50 FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_adam_apply_one_fusion", "before"); in TEST_F()
63 FuncGraphPtr new_graph = optimizer->Optimize(fg); in TEST_F()
65 FuncGraphPtr g_after = get_py_fun_.CallAndParseRet("test_adam_apply_one_fusion", "after"); in TEST_F()
88 FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_adam_apply_one_fusion", "before_cond1"); in TEST_F()
101 FuncGraphPtr new_graph = optimizer->Optimize(fg); in TEST_F()
103 FuncGraphPtr g_after = get_py_fun_.CallAndParseRet("test_adam_apply_one_fusion", "after"); in TEST_F()
126 FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_adam_apply_one_fusion", "before_cond2"); in TEST_F()
139 FuncGraphPtr new_graph = optimizer->Optimize(fg); in TEST_F()
141 FuncGraphPtr g_after = get_py_fun_.CallAndParseRet("test_adam_apply_one_fusion", "after"); in TEST_F()
164 FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_adam_apply_one_fusion", "before_cond3"); in TEST_F()
[all …]
Dadam_apply_one_with_decay_rule_test.cc34 …FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_adam_apply_one_with_decay_rule", "before_cond1"… in TEST_F()
48 FuncGraphPtr new_graph = optimizer->Optimize(fg); in TEST_F()
50 …FuncGraphPtr g_after = get_py_fun_.CallAndParseRet("test_adam_apply_one_with_decay_rule", "after"); in TEST_F()
55 …FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_adam_apply_one_with_decay_rule", "before_cond2"… in TEST_F()
69 FuncGraphPtr new_graph = optimizer->Optimize(fg); in TEST_F()
71 …FuncGraphPtr g_after = get_py_fun_.CallAndParseRet("test_adam_apply_one_with_decay_rule", "after"); in TEST_F()
76 …FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_adam_apply_one_with_decay_rule", "before_cond3"… in TEST_F()
90 FuncGraphPtr new_graph = optimizer->Optimize(fg); in TEST_F()
92 …FuncGraphPtr g_after = get_py_fun_.CallAndParseRet("test_adam_apply_one_with_decay_rule", "after"); in TEST_F()
97 …FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_adam_apply_one_with_decay_rule", "before_cond4"… in TEST_F()
[all …]
Dlamb_next_mv_with_decay_rule_test.cc58 FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_lamb_next_mv_with_decay_rule_cond4", "before"); in TEST_F()
71 FuncGraphPtr new_graph = optimizer->Optimize(fg); in TEST_F()
73 …FuncGraphPtr g_after = get_py_fun_.CallAndParseRet("test_lamb_next_mv_with_decay_rule_cond4", "aft… in TEST_F()
102 …FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_lamb_next_mv_with_decay_rule_cond4", "before_un… in TEST_F()
116 FuncGraphPtr new_graph = optimizer->Optimize(fg); in TEST_F()
118 …FuncGraphPtr g_after = get_py_fun_.CallAndParseRet("test_lamb_next_mv_with_decay_rule_cond4", "aft… in TEST_F()
147 …FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_lamb_next_mv_with_decay_rule_cond4", "before_un… in TEST_F()
161 FuncGraphPtr new_graph = optimizer->Optimize(fg); in TEST_F()
163 …FuncGraphPtr g_after = get_py_fun_.CallAndParseRet("test_lamb_next_mv_with_decay_rule_cond4", "aft… in TEST_F()
192 …FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_lamb_next_mv_with_decay_rule_cond4", "before_un… in TEST_F()
[all …]
Dlamb_next_mv_rule_test.cc57 FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_lamb_next_mv_rule_cond4", "before"); in TEST_F()
70 FuncGraphPtr new_graph = optimizer->Optimize(fg); in TEST_F()
72 FuncGraphPtr g_after = get_py_fun_.CallAndParseRet("test_lamb_next_mv_rule_cond4", "after"); in TEST_F()
100 …FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_lamb_next_mv_rule_cond4", "before_unmatched_rea… in TEST_F()
114 FuncGraphPtr new_graph = optimizer->Optimize(fg); in TEST_F()
143 …FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_lamb_next_mv_rule_cond4", "before_unmatched_rea… in TEST_F()
157 FuncGraphPtr new_graph = optimizer->Optimize(fg); in TEST_F()
186 …FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_lamb_next_mv_rule_cond4", "before_unmatched_rea… in TEST_F()
200 FuncGraphPtr new_graph = optimizer->Optimize(fg); in TEST_F()
229 …FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_lamb_next_mv_rule_cond4", "before_unmatched_rea… in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/debug/
Danf_ir_utils.h74 void ExportFuncGraph(const std::string &filename, const FuncGraphPtr &func_graph);
78 …int GetParamIndex(const FuncGraphPtr &func_graph, const AnfNodePtr &param, bool throw_excp = true);
80 std::string GetValueNodeText(const FuncGraphPtr &func_graph, const ValueNodePtr &node);
82 …std::string GetSymbolicKeyInstanceText(const FuncGraphPtr &func_graph, const SymbolicKeyInstancePt…
83 std::string GetSequenceText(const FuncGraphPtr &func_graph, const ValuePtr &value);
84 std::string GetValueText(const FuncGraphPtr &func_graph, const ValuePtr &value);
85 std::string GetOtherValueText(const FuncGraphPtr &func_graph, const ValuePtr &value);
87 std::string GetDictText(const FuncGraphPtr &func_graph, const ValuePtr &value);
90 std::string GetAnfNodeText(const FuncGraphPtr &func_graph, const AnfNodePtr &node,
96 void OutputOrderList(std::ofstream &ofs, const FuncGraphPtr &func_graph);
[all …]
/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Doptimize_dependence_test.cc40 FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_optimize_dependence", "before"); in TEST_F()
46 FuncGraphPtr new_graph = optimizer->Optimize(g); in TEST_F()
48 FuncGraphPtr g_after = get_py_fun_.CallAndParseRet("test_optimize_dependence", "after"); in TEST_F()
60 …FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_optimize_dependence_with_make_tuple", "before"); in TEST_F()
66 FuncGraphPtr new_graph = optimizer->Optimize(g); in TEST_F()
68 …FuncGraphPtr g_after = get_py_fun_.CallAndParseRet("test_optimize_dependence_with_make_tuple", "af… in TEST_F()
81 …FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_optimize_control_dependence_with_make_tuple", "… in TEST_F()
87 FuncGraphPtr new_graph = optimizer->Optimize(g); in TEST_F()
89 …FuncGraphPtr g_after = get_py_fun_.CallAndParseRet("test_optimize_control_dependence_with_make_tup… in TEST_F()
102 FuncGraphPtr g = get_py_fun_.CallAndParseRet("test_optimize_control_dependence", "before"); in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/
Dcomposite.h70 FuncGraphPtr GenerateFromTypes(const TypePtrList &args_spec_list) override;
74 …AnfNodePtr FullMake(const FuncGraphPtr &func_graph, const AnfNodePtr &fn_arg, const ArgsPairList &…
75 …AnfNodePtr FullMake(const std::shared_ptr<List> &type, const FuncGraphPtr &func_graph, const AnfNo…
77 …AnfNodePtr FullMake(const std::shared_ptr<Tuple> &type, const FuncGraphPtr &func_graph, const AnfN…
79 …AnfNodePtr FullMake(const std::shared_ptr<Class> &type, const FuncGraphPtr &func_graph, const AnfN…
81 AnfNodePtr Make(const FuncGraphPtr &graph, const AnfNodePtr &fn_arg, const ArgsPairList &arg_map);
82 ArgsPairList Harmonize(const FuncGraphPtr &graph, const ArgsPairList &args_spec_list);
111 FuncGraphPtr GenerateFuncGraph(const AbstractBasePtrList &args_spec_list) override;
112 FuncGraphPtr GenerateSequeueFuncGraph(const abstract::AbstractSequeuePtr &sequeue) const;
128 FuncGraphPtr GenerateFuncGraph(const AbstractBasePtrList &args_spec_list) override;
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dinline.h46 auto fg = GetValueNode<FuncGraphPtr>(node); in operator()
77 … (IsValueNode<FuncGraph>(inner) && GetValueNode<FuncGraphPtr>(inner)->parent() == nullptr)) { in operator()
87 using CriterionFuncType = std::function<bool(InlinerBase *, const FuncGraphPtr &, const AnfNodePtr …
89 bool IsUniqueUse(InlinerBase *, const FuncGraphPtr &fg, const AnfNodePtr &);
91 bool IsTrivial(InlinerBase *, const FuncGraphPtr &fg, const AnfNodePtr &);
92 bool IsInside(InlinerBase *, const FuncGraphPtr &, const AnfNodePtr &node);
93 bool IsCore(InlinerBase *, const FuncGraphPtr &fg, const AnfNodePtr &);
94 bool IsDirectParentCall(InlinerBase *, const FuncGraphPtr &fg, const AnfNodePtr &node);
95 bool IsNotRecursive(InlinerBase *inliner, const FuncGraphPtr &fg, const AnfNodePtr &);
96 bool IsForceInline(InlinerBase *, const FuncGraphPtr &fg, const AnfNodePtr &) { in IsForceInline()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.h35 void HandleReceiveParam(const FuncGraphPtr &root, const std::vector<AnfNodePtr> &all_nodes);
36 void AddVirtualAssignAdd(const FuncGraphPtr &root);
39 const FuncGraphPtr &root);
42 const PipelinePair &forward_end_before_pair, const FuncGraphPtr &root);
46 const PipelinePair &forward_start_pair, const FuncGraphPtr &root);
49 const FuncGraphPtr &root);
50 PipelinePair Deduplicate(const std::vector<AnfNodePtr> &node_vector, const FuncGraphPtr &root, int6…
55 std::vector<AnfNodePtr> *allreduce_params, const FuncGraphPtr &root);
56 void Reorder(const FuncGraphPtr &root);
57 void ReorderForPredict(const FuncGraphPtr &root, const FuncGraphManagerPtr &manager);
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.h61 const FuncGraphPtr &func_graph, int64_t pos, const CNodePtr &pre_node);
74 void MarkForwardCNode(const FuncGraphPtr &root);
85 std::pair<AnfNodePtr, bool> FindParameter(const AnfNodePtr &node, const FuncGraphPtr &func_graph);
87 …odePtr> FindCNode(const AnfNodePtr &anode, const std::string &name, const FuncGraphPtr &func_graph,
105 std::pair<AnfNodePtr, int64_t> FindSubGraph(const FuncGraphPtr &func_graph, const AnfNodePtr &param…
111 void CoverSliceShape(const FuncGraphPtr &root);
114 bool IsInsertVirtualOutput(const FuncGraphPtr &root);
133 void ParallelCommunication(const FuncGraphPtr &root, const std::vector<AnfNodePtr> &all_nodes,
138 void CheckpointStrategy(const std::vector<AnfNodePtr> &all_nodes, const FuncGraphPtr &root);
141 bool StepParallel(const FuncGraphPtr &func_graph, const opt::OptimizerPtr &optimizer);
[all …]

12345678910>>...37