AVh dZddlmZddlmZddlmZddlm Z e dgdZ e d gd Z e d gd Z y )z!API for enabling v2 control flow.)ops)control_flow_util) tf_logging) tf_exportenable_control_flow_v2)v1ctjddtjj j ddt _y)aUse control flow v2. 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.while_loop` and `tf.cond` will generate functional `While` and `If` ops instead of low-level `Switch`, `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. Note: v2 control flow is always enabled inside of tf.function. Calling this function is not required. zEnabling control flow v2TNloggingvlogr_control_flow_api_gaugeget_cellsetrENABLE_CONTROL_FLOW_V2]/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/control_flow_v2_toggles.pyrrs9$ ,,q,-&&(,,T2-1*rdisable_control_flow_v2ctjddtjj j ddt _y)zOpts out of control flow v2. Note: v2 control flow is always enabled inside of tf.function. Calling this function has no effect in that case. If your code needs tf.disable_control_flow_v2() to be called to work properly please file a bug. r zDisabling control flow v2FNr rrrrr/s9 ,,q-.&&(,,U3-2*rcontrol_flow_v2_enabledcPtjtjS)zrReturns `True` if v2 control flow is enabled. Note: v2 control flow is always enabled inside of tf.function. )rEnableControlFlowV2rget_default_graphrrrrr?s  . .s/D/D/F GGrN) __doc__tensorflow.python.frameworkrtensorflow.python.opsrtensorflow.python.platformrr tensorflow.python.util.tf_exportrrrrrrrr so (+3<6 '()2*2, ()* 3+ 3 ()*H+Hr