tf.contrib.autograph.set_loop_options

tf.contrib.autograph.set_loop_options(
    parallel_iterations=UNSPECIFIED,
    back_prop=UNSPECIFIED,
    swap_memory=UNSPECIFIED,
    maximum_iterations=UNSPECIFIED
)

Defined in tensorflow/python/autograph/lang/directives.py.

Specifies additional arguments to be passed to the enclosing while_loop.

The parameters apply to and only to the immediately enclosing loop. It only has effect if the loop is staged as a TF while_loop; otherwise the parameters have no effect.

Args:

  • parallel_iterations: See tf.while_loop.
  • back_prop: See tf.while_loop.
  • swap_memory: See tf.while_loop.
  • maximum_iterations: See tf.while_loop.