Searched refs:OptPassGroupMap (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pass.cc | 58 using OptPassGroupMap = opt::OptPassGroupMap; typedef
132 OptPassGroupMap map( in PrimBpOptPassStep1()
160 OptPassGroupMap map({{"ad_renormalize", opt::OptPassConfig::Renormalize()}, in PrimBpOptPassStep2()
190 OptPassGroupMap map({ in BpropGraphFinalOptPass()
227 void AddParallelRenormalize(OptPassGroupMap *map_a) { in AddParallelRenormalize()
291 OptPassGroupMap GetOptPassesA(const opt::irpass::OptimizeIRPassLib &irpass) { in GetOptPassesA()
337 OptPassGroupMap map_a({{"switch_simplify", opt::OptPassConfig({irpass.switch_simplify_})}, in GetOptPassesA()
362 OptPassGroupMap GetA1A2(const opt::irpass::OptimizeIRPassLib &irpass) { in GetA1A2()
365 OptPassGroupMap a1_a2(opt_a.begin(), opt_a.begin() + a1_a2_len); in GetA1A2()
369 OptPassGroupMap GetOptPassesAfterCconv(const opt::irpass::OptimizeIRPassLib &irpass) { in GetOptPassesAfterCconv()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | optimizer.h | 90 using OptPassGroupMap = std::vector<std::pair<std::string, OptPassConfig>>; variable
104 void Init(const OptPassGroupMap &passes, bool run_only_once) { in Init()
135 … const OptPassGroupMap &passes, bool run_only_once = false,
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | grad.cc | 33 opt::OptPassGroupMap map({{"partial_eliminate_", partial_eliminate_opt_}}); in PartialEliminateOptPass()
|
| D | kpynative.cc | 63 opt::OptPassGroupMap map({{"eliminate_zeros_like_prim_", eliminate_zeros_like_prim_pass}}); in ZerosLikePrimOptPass()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/ |
| D | resolve.cc | 363 opt::OptPassGroupMap GetOptResolvePasses(const opt::irpass::ResolveIRPassLib &irpass) { in GetOptResolvePasses()
365 opt::OptPassGroupMap map({ in GetOptResolvePasses()
|