Searched refs:OptPassConfig (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pass.cc | 120 opt::OptPassConfig pynative_eliminate = opt::OptPassConfig({ in PrimBpOptPassStep1()
124 opt::OptPassConfig switch_simplify = opt::OptPassConfig({ in PrimBpOptPassStep1()
128 opt::OptPassConfig inline_opt = opt::OptPassConfig({ in PrimBpOptPassStep1()
146 opt::OptPassConfig special_op_simplify = opt::OptPassConfig({ in PrimBpOptPassStep2()
153 opt::OptPassConfig inline_opt = opt::OptPassConfig({ in PrimBpOptPassStep2()
160 OptPassGroupMap map({{"ad_renormalize", opt::OptPassConfig::Renormalize()}, in PrimBpOptPassStep2()
163 {"auto_monad_grad", opt::OptPassConfig(re_auto_monadwrapper)}}); in PrimBpOptPassStep2()
179 opt::OptPassConfig bg_final_opt = opt::OptPassConfig({ in BpropGraphFinalOptPass()
189 opt::OptPassConfig fill_zeros_like = opt::OptPassConfig{irpass.zero_like_fill_zero_}; in BpropGraphFinalOptPass()
196 (void)map.emplace_back(std::make_pair("renormalize", opt::OptPassConfig::Renormalize())); in BpropGraphFinalOptPass()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | optimizer.h | 43 class OptPassConfig {
45 explicit OptPassConfig(const OptimizeGraphFunc &func) : func_(func) {} in OptPassConfig() function
46 …explicit OptPassConfig(const std::vector<SubstitutionPtr> &list, bool is_once = false, bool global…
48 …OptPassConfig(const std::initializer_list<SubstitutionPtr> &list, bool is_once = false, bool globa…
50 ~OptPassConfig() = default;
55 static OptPassConfig Renormalize() { return OptPassConfig(); } in Renormalize()
63 OptPassConfig() : is_renormalize_(true) {} in OptPassConfig() function
90 using OptPassGroupMap = std::vector<std::pair<std::string, OptPassConfig>>;
114 const OptPassConfig &config = iter.second; in Init()
|
| /third_party/mindspore/tests/ut/cpp/optimizer/ |
| D | optimizer_test.cc | 57 {"grad", opt::OptPassConfig(opt::irpass::ExpandJPrim())}, in TEST_F()
58 {"cse", OptPassConfig(CSEPass(false))}}, in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | grad.cc | 31 opt::OptPassConfig partial_eliminate_opt_ = opt::OptPassConfig( in PartialEliminateOptPass()
|
| D | kpynative.cc | 59 opt::OptPassConfig eliminate_zeros_like_prim_pass = opt::OptPassConfig({ in ZerosLikePrimOptPass()
|