Searched defs:run_only_once (Results 1 – 5 of 5) sorted by relevance
41 def __init__(self, requires_grad=True, run_only_once=False): argument90 def register_pass(requires_grad=True, run_only_once=False): argument
51 void SetRunOnlyOnce(bool run_only_once) { run_only_once_ = run_only_once; } in SetRunOnlyOnce()
52 bool requires_grad, bool run_only_once) { in Register()
104 void Init(const OptPassGroupMap &passes, bool run_only_once) { in Init()
85 FuncGraphPtr GraphOptimizer::Optimize(const FuncGraphPtr &func_graph, bool run_only_once) { in Optimize()