Exemplo n.º 1
0
 def cond(args):
     i, lower, _, upper = args
     return jnp.logical_and(jnp.less(i, max_it),
                            jnp.less(abs_tol, jnp.amax(upper - lower)))
Exemplo n.º 2
0
 def cond(iqc):
     i, q, count = iqc
     return jnp.less(i, n)