BVh1dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z dd l mZejd d d Zej"dZdZedgdZedgdZy)z!Switching v2 features on and off.)tf2) monitoring)ops)registry)tensor) tensor_shape)control_flow_v2_toggles)resource_variables_toggle) tf_exportz/tensorflow/version/v2_behaviorz*whether v2_behavior is enabled or disabledstatusdata_v2_callbackscDtj||jy)N)_DATA_V2_CALLBACKSregister __module__) data_v2_funcs R/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/compat/v2_compat.pyregister_data_v2_callbackr#slL,C,CDenable_v2_behavior)v1ctjdjdtjt j tjtjtjtjtj!D]}tj#|}| y)a[Enables TensorFlow 2.x behaviors. This function can be called at the beginning of the program (before `Tensors`, `Graphs` or other structures have been created, and before devices have been initialized. It switches all global behaviors that are different between TensorFlow 1.x and 2.x to behave as intended for 2.x. This function is called in the main TensorFlow `__init__.py` file, user should not need to call it, except during complex migrations. @compatibility(TF2) This function is not necessary if you are using TF2. V2 behavior is enabled by default. @end_compatibility enableTN)_v2_behavior_usage_gaugeget_cellsetrrrenable_eager_executionrenable_v2_tensorshaper enable_resource_variablesrenable_tensor_equalityr enable_control_flow_v2rlistlookup)v2_enabler_name v2_enablers rrr's"##H-11$7**,$$&557!002+002o#**?;JLrdisable_v2_behaviorctjdjdtjt j tjtjtjtjtj!D]}tj#|}| y)a Disables TensorFlow 2.x behaviors. This function can be called at the beginning of the program (before `Tensors`, `Graphs` or other structures have been created, and before devices have been initialized. It switches all global behaviors that are different between TensorFlow 1.x and 2.x to behave as intended for 1.x. User can call this function to disable 2.x behavior during complex migrations. @compatibility(TF2) Using this function indicates that your software is not compatible with eager execution and `tf.function` in TF2. To migrate to TF2, rewrite your code to be compatible with eager execution. Please refer to the [migration guide] (https://www.tensorflow.org/guide/migrate) for additional resource on the topic. @end_compatibility disableTN)rrrrr(rdisable_eager_executionrdisable_v2_tensorshaper disable_resource_variablesrdisable_tensor_equalityr disable_control_flow_v2rr"r#)v2_disabler_name v2_disablers rr&r&Is*##I.2248++-%%'668  "113,113$++,<=KMrN)__doc__tensorflow.pythonrtensorflow.python.eagerrtensorflow.python.frameworkrrrrtensorflow.python.opsr r tensorflow.python.util.tf_exportr BoolGaugerRegistryrrrr&rrr9s(!.+0.49;60://%0(<'X&&':;E #$%&B $%&'r