Lines Matching full:control_flow
18 import executorch.exir.control_flow as control_flow namespace
50 from functorch.experimental.control_flow import cond
282 @control_flow.tracing_context(
288 @control_flow.tracing_context(
293 @control_flow.tracing_context(
299 @control_flow.tracing_context(
312 y, _ = control_flow.while_loop(
328 @control_flow.tracing_context(
332 @control_flow.tracing_context(
341 @control_flow.tracing_context(
350 return control_flow.while_loop(loop_cond, loop_body, (accum, cnt))
352 @control_flow.tracing_context(