Home
last modified time | relevance | path

Searched refs:lower_control_flow (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dconvert_to_constants.py796 lower_control_flow, argument
817 graph_def = _run_inline_graph_optimization(func, lower_control_flow,
887 lower_control_flow, argument
917 lower_control_flow,
990 def _run_inline_graph_optimization(func, lower_control_flow, argument
1009 if not lower_control_flow:
1129 lower_control_flow=True, argument
1156 lower_control_flow=lower_control_flow,
1167 lower_control_flow=True, argument
1202 lower_control_flow=lower_control_flow,
[all …]
H A Dconvert_to_constants_test.py177 input_func, lower_control_flow=False)
546 input_func, lower_control_flow=False)
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dlower_functional_ops.cc153 const auto lower_control_flow = [&](Node* node) -> bool { in Run() local
181 if (n->IsIfNode() && lower_control_flow(n)) { in Run()
184 } else if (n->IsCaseNode() && lower_control_flow(n)) { in Run()
187 } else if (n->IsWhileNode() && lower_control_flow(n)) { in Run()
192 DCHECK(!lower_control_flow(n)) in Run()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dfunction_optimizer.h30 bool lower_control_flow) in FunctionOptimizer() argument
31 : opt_level_(opt_level), lower_control_flow_(lower_control_flow) {} in FunctionOptimizer()
H A Dfunction_optimizer.cc1211 const bool lower_control_flow, in InlineFunctionCalls() argument
1259 if (lower_control_flow && LowerUsingSwitchMergeIsOn(n)) { in InlineFunctionCalls()
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dlite.py1304 self._funcs[0], lower_control_flow=False))
1402 self._funcs[0], lower_control_flow=False))
2278 concrete_func, lower_control_flow=False)