|View source on GitHub|
Use control flow v2.
Compat aliases for migration
See Migration guide for more details.
control flow v2 (cfv2) is an improved version of control flow in TensorFlow
with support for higher order derivatives. Enabling cfv2 will change the
graph/function representation of control flow, e.g.,
tf.cond will generate functional
If ops instead of low-level
Merge etc. ops. Note: Importing and running graphs exported
with old control flow will still be supported.
Calling tf.enable_control_flow_v2() lets you opt-in to this TensorFlow 2.0 feature.