BVh'dZddlZddlmZddlmZddlmZddlm Z ddlm Z ddl m Z ejZd Zej d d Zd Zd ZdZdZGddeZy)z(Provides wrapper for TensorFlow modules.N) monitoring) tf_logging)fast_module_type) tf_decorator) tf_inspect)all_renames_v2z/tensorflow/api/compat/v1zcompat.v1 usagecN|tjvrytj|SN)rsymbol_renames)names U/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/util/module_wrapper.py get_rename_v2r s$ ...   & &t ,,c"tj}|jjjdk(sJdt dD]}|j}|n|}dj |jj|jS)z|Extracts the caller filename and line number as a string. Returns: A string describing the caller source location. _tfmw_add_deprecation_warningzThis function should be called directly from _tfmw_add_deprecation_warning, as the caller is identified heuristically by chopping off the top stack frames.z{}:{}) r currentframef_backf_codeco_namerangeformat co_filenamef_lineno)frame_parents r_call_locationr&s  ! ! #%    $ $(G G=<= G 8a \\F ~ E   00%.. AArc&td|DS)Nc3:K|]}|jdk(yw) deprecatedN)decorator_name).0ds r z1contains_deprecation_decorator..>s B!Q   - Bs)any) decoratorss rcontains_deprecation_decoratorr)=s Bz B BBrctj|\}}t|rytj|rytj |syt |dsytj|j\}}t|S)a%Checks if given object has a deprecation decorator. We check if deprecation decorator is in decorators as well as whether symbol is a class whose __init__ method has a deprecation decorator. Args: symbol: Python object. Returns: True if symbol has deprecation decorator. TF__init__)runwrapr)r isfunctionisclasshasattrr+)symbolr(init_decoratorsrs rhas_deprecation_decoratorr2Asw$**62*f#J/ 6"    F #   $ #**6??;/1 ' 88rcleZdZdZdZ d fd ZdZdZdZdZ fdZ d Z fd Z d Z d ZxZS)TFModuleWrapperzGWrapper for TF modules to support deprecation messages and lazyloading.Fctt| |jtj |tj tj|tj|jj|j||_ ||_ ||_ ||_||_|jj!d|_|jr[t%|jj'|j_t%|jj'|_nt+|jdr|jj(|_n^t-|jDcgc]}|j/ds|c}|j_|jj(|_t1|_d|_ycc}w)Nz .compat.v1__all__rr)superr4r+__name__FastModuleTypeset_getattr_callback_getattrset_getattribute_callback _getattribute__dict__update_tfmw_wrapped_module_tfmw_module_name_tfmw_public_apis _tfmw_print_deprecation_warnings_tfmw_has_liteendswith_tfmw_is_compat_v1listkeysr6r/dir startswithset_tfmw_deprecated_checked_tfmw_warning_count)selfwrapped module_name public_apis deprecationhas_liteattr __class__s rr+zTFModuleWrapper.__init___sv /4)'*:*:;''o.F.FG,,T-<-J-JLMM))*!(D(D(D,7D)"D&//88FD *.t/E/E/J/J/L*Md'$005578dl **I 60088 !!:!:;- ??3' - !!)0088 %(ED! D- s?Gc|jtkr||jvr|jj||jrd|jd|}nd|z}t |}|r\t |sQt}|jds6tjdt|||xjdz c_yy) z@Print deprecation warning for attr with given name if necessary.ztf..ztf.%stjs.dt_tj j d|j |}|dr(tj|d}t||d}ntj|d}t|j||||j|<|j|||S)zLazily loading the modules.appTrr )rFr4compat_v1_usage_recordedcompat_v1_usage_gaugeget_cellrKrB importlib import_modulegetattrsetattrr@r>_fastdict_insert)rNr symbol_loc_infomodulerTs r_tfmw_import_modulez#TFModuleWrapper._tfmw_import_modules DEM  4 415o.$$&**40,,T2Oq&&q'9:f V_Q/ 0d  $ $_Q%7 8d D % %tT2DMM$$% Krctj|d}|dk(r?|jr3|j|}t |j d|||||Stj||}|j ds"|j ds|j dr ||||S|jr|j||s ||||S)zImports and caches pre-defined API. Warns if necessary. This method is a replacement for __getattribute__(). It will be added into the extended python module as a callback to reduce API overhead. rilite___tfmw_ _fastdict_) object__getattribute__rDrlrhr@rJrCr)rNr func__fastdict_insertrTs rr=zTFModuleWrapper._getattributes#33D:LM v~  ''-))648dD)   " "4 .D  t 9T__>D$' k  1 1  . .tT :D$' Krc t|j|}|j r|j |||S#t$r0|js||jvr|j |}YXwxYw)aImports and caches pre-defined API. Warns if necessary. This method is a replacement for __getattr__(). It will be added into the extended python module as a callback to reduce API overhead. Instead of relying on implicit AttributeError handling, this added callback function will be called explicitly from the extended C API if the default attribute lookup fails. )rgr@AttributeErrorrBrlrCr)rNr rTs rr;zTFModuleWrapper._getattrsy , T.. 5d ,, ((t4 K , # # T++ +  % %d +d,s86A10A1c>|jdswt|j||||j|<||jvr |dk7r|jj ||j |r|j||tt|+||y)Nrpr6) rJrhr@r>r6append_fastdict_key_inrir7r4 __setattr__)rNargvalrUs rrzzTFModuleWrapper.__setattr__s >>( # d''c2dmmC DLL SI%5 C  s # c3' /4,S#6rc :|jrvtt|jjj tt |j Dcgc]}|jds|c}St |j Scc}w)Nr)rBrGrKrHunionrIr@rJ)rNrTs r__dir__zTFModuleWrapper.__dir__s   d$$))+ , 2 2#&t'@'@#A- ** ++ sBc&|jdrtt||yt |j ||j j|||jvr|jj||j|y)Nrp) rJr7r4 __delattr__delattrr@r>popr6remove _fastdict_pop)rNr rUs rrzTFModuleWrapper.__delattr__ sl x  OT.t4 d''. mm   D! rc6|jjSr )r@__repr__rNs rrzTFModuleWrapper.__repr__s  $ $ - - //rc<tj|jffSr )rerfr8rs r __reduce__zTFModuleWrapper.__reduce__s  " "T]]$4 44r)NTF)r8 __module__ __qualname____doc__rbr+rrlr=r;rzrrrr __classcell__)rUs@rr4r4ZsKO"  $!L.,&P8 7 ,05rr4)rretensorflow.python.eagerrtensorflow.python.platformrr[tensorflow.python.utilrrrtensorflow.tools.compatibilityrget_fast_module_type_classr9rY BoolGaugercrrr)r2r4rrrst/.<3/-9<!<<>, ,,-H->@- B.C92A5nA5r