def while_compute(x):
return control_flow_ops.while_loop_v2(
lambda i, _: i < 10, lambda i, y:
(i + 1, y + vectorized_compute(x, i)),
(0, array_ops.zeros([5, 1])))[1]
def g(x):
return control_flow_ops.while_loop_v2(
lambda *_: True,
lambda y, shp: (y + random_ops.random_normal(shp)**2, shp),
(x, array_ops.shape(x)),
maximum_iterations=3)[0]
def outer(y, shp):
y, shp = control_flow_ops.while_loop_v2(
lambda *_: True, inner, (y, shp), maximum_iterations=3)
y, shp = array_ops.identity_n([y, shp])
return control_flow_ops.while_loop_v2(
lambda *_: True, inner, (y, shp), maximum_iterations=5)
def outer(y, shp):
y, shp = control_flow_ops.while_loop_v2(
lambda *_: True, inner, (y, shp), maximum_iterations=3)
return control_flow_ops.while_loop_v2(
lambda *_: True, inner, (y, shp), maximum_iterations=4)