BVh4dZddlmZddlmZdZddZdZy) zContains keras-specific LossScale functionality. This functions cannot be in the non-keras loss_scale.py file since they depend on keras, and files outside of keras should not depend on files inside keras. ) generic_utils loss_scalec,tj|SN)rserialize_keras_objectrs b/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/keras/mixed_precision/loss_scale.py serializer s  - -j 99Ncvtjtjd}tj|||dS)N)FixedLossScaleDynamicLossScalez loss scale)module_objectscustom_objectsprintable_module_name)loss_scale_moduler rrdeserialize_keras_object)configrloss_scale_module_objectss r deserializers=)88+<<  / / .#(  r ct|tr t|St|ttfrt j |S|dk(rt jSt|t jr|S|ytd|z)zGet a loss scale object.dynamicNz-Could not interpret loss scale identifier: %s) isinstancedictrintfloatrr r LossScale ValueError) identifiers r getr +s D! z "" S%L)  + +J 779  - - // -778   D  !!r r) __doc__tensorflow.python.keras.utilsr'tensorflow.python.training.experimentalrrr rr r r r%s! 8S: !r