Home
last modified time | relevance | path

Searched refs:run_only_once (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dpass_group.h31 explicit PassGroup(const std::string &name = "pass_group", bool run_only_once = false)
32 : name_(name), passes_{}, run_only_once_(run_only_once) {} in name_()
51 void SetRunOnlyOnce(bool run_only_once) { run_only_once_ = run_only_once; } in SetRunOnlyOnce() argument
Dpy_pass_manager.cc52 bool requires_grad, bool run_only_once) { in Register() argument
60 cur_pg->SetRunOnlyOnce(run_only_once); in Register()
64 PythonPassPtr new_pass = std::make_shared<PythonPass>(pass_name, pattern, target, run_only_once); in Register()
Dpy_pass.h37 …(const std::string &name, const PatternPtr &src, const PatternPtr &dst, bool run_only_once = false)
38 : src_pattern_(src), dst_pattern_(dst), name_(name), run_only_once_(run_only_once) {} in src_pattern_()
Doptimizer.h104 void Init(const OptPassGroupMap &passes, bool run_only_once) { in Init() argument
105 run_only_once_ = run_only_once; in Init()
135 … const OptPassGroupMap &passes, bool run_only_once = false,
138 optimizer->Init(passes, run_only_once);
Dpy_pass_manager.h56 bool run_only_once);
/third_party/mindspore/tests/ut/python/optimizer/
Dtest_python_pass.py48 @register_pass(run_only_once=True)
64 @register_pass(run_only_once=True)
87 @register_pass(run_only_once=True)
112 @register_pass(run_only_once=True)
136 @register_pass(run_only_once=True)
177 @register_pass(requires_grad=False, run_only_once=True)
189 @register_pass(requires_grad=False, run_only_once=True)
213 @register_pass(run_only_once=True)
240 @register_pass(requires_grad=False, run_only_once=True)
264 @register_pass(requires_grad=False, run_only_once=True)
[all …]
/third_party/mindspore/mindspore/graph_utils/python_pass/
Dpython_pass_register.py41 def __init__(self, requires_grad=True, run_only_once=False): argument
44 if not isinstance(run_only_once, bool):
47 self.run_only_once_ = run_only_once
90 def register_pass(requires_grad=True, run_only_once=False): argument
112 return PyPassManager(requires_grad, run_only_once)
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dpass_manager.h47 explicit PassManager(const std::string &name = "pm", bool run_only_once = true)
48 …: name_(name), passes_{}, run_only_once_(run_only_once), cache_manager_(std::make_shared<CacheMana… in name_()
Doptimizer.cc85 FuncGraphPtr GraphOptimizer::Optimize(const FuncGraphPtr &func_graph, bool run_only_once) { in Optimize() argument
87 run_only_once_ = (pass_managers_.size() == 1) ? true : run_only_once; in Optimize()
Doptimizer.h77 FuncGraphPtr Optimize(const FuncGraphPtr &func_graph, bool run_only_once = true);