Lines Matching refs:scf

6 from mlir.dialects import scf
27     loop = scf.ForOp(lb, ub, step, [lb, lb])
29       scf.YieldOp(loop.inner_iter_args)
46     loop = scf.ForOp(lb, ub, step, [lb])
48       scf.YieldOp([loop.induction_variable])
68     loop = scf.ForOp(lb, ub, step, iter_args)
70       scf.YieldOp(loop.inner_iter_args)
94     if_op = scf.IfOp(cond)
98       scf.YieldOp([])
116     if_op = scf.IfOp(cond, [i32, i32], hasElse=True)
120       scf.YieldOp([x_true, y_true])
124       scf.YieldOp([x_false, y_false])