BVh. lUdZddlmZddlZddlZddlmZmZmZm Z m Z ddl m Z ddl m Z dZdZgZGd d eZeed d eed diZeeddeeddiZGddeZeZeeefed<dedeefdZedfdedededeefdZdeedeedeefd Zdedeefd!Zdedeefd"Zd#edeefd$Z d#edeefd%Z!e d&Z"Gd'd(e#Z$d)edefd*Z%Gd+d,e Z&ejNe$e-Z(e&ed.<ejNe$e-Z)e&ed/<y)0aUtilities for exporting TensorFlow symbols to the API. Exporting a function or a class: To export a function or a class use tf_export decorator. For e.g.: ```python @tf_export('foo', 'bar.foo') def foo(...): ... ``` If a function is assigned to a variable, you can export it by calling tf_export explicitly. For e.g.: ```python foo = get_foo(...) tf_export('foo', 'bar.foo')(foo) ``` Exporting a constant ```python foo = 1 tf_export('consts.foo').export_constant(__name__, 'foo') ``` )SequenceN)Any NamedTupleOptionalProtocolTypeVar) tf_decorator) tf_inspectkeras tensorflowc"eZdZUeed<eed<y) _Attributesnames constantsN)__name__ __module__ __qualname__str__annotations__P/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/util/tf_export.pyrr8s * .rr _tf_api_names_tf_api_constants_keras_api_names_keras_api_constants_tf_api_names_v1_tf_api_constants_v1_keras_api_names_v1_keras_api_constants_v1ceZdZdZy)InvalidSymbolNameErrorzDRaised when trying to export symbol as an invalid or unallowed name.N)rrr__doc__rrrr"r"MsLrr"_NAME_TO_SYMBOL_MAPPINGnamereturnc,tj|SN)r$get)r%s rget_symbol_from_namer*Ts $ $T **rFsymbolapi_nameadd_prefix_to_v1_namescht|dsyt|j}tj|\}}||j vryt ||}|j jdg}t||}|r|St|j}t ||}t||} |rd| zS| S)aGet canonical name for the API symbol. Example: ```python from tensorflow.python.util import tf_export cls = tf_export.get_symbol_from_name('keras.optimizers.Adam') # Gives `` print(cls) # Gives `keras.optimizers.Adam` print(tf_export.get_canonical_name_for_symbol(cls, api_name='keras')) ``` Args: symbol: API function or class. api_name: API name. Currently, only `tensorflow`. add_prefix_to_v1_names: Specifies whether a name available only in V1 should be prefixed with compat.v1. Returns: Canonical name for the API symbol (for e.g. initializers.zeros) if canonical name could be determined. Otherwise, returns None. __dict__N_tf_deprecated_api_names compat.v1.%s) hasattr API_ATTRSrr unwrapr/getattrr)get_canonical_name API_ATTRS_V1) r+r,r-api_names_attr_undecorated_symbol api_namesdeprecated_api_namescanonical_namev1_canonical_names rget_canonical_name_for_symbolr?Xs:  $ X&,,.&--f5!-666 (.9)+4488 "&i1EF.  )//.(.9)(4HI - -- rr;r<cFtfd|Dd}|r|S|r|dSy)aGet preferred endpoint name. Args: api_names: API names iterable. deprecated_api_names: Deprecated API names iterable. Returns: Returns one of the following in decreasing preference: - first non-deprecated endpoint - first endpoint - None c3,K|] }|vs| ywr(r).0r%r<s r z%get_canonical_name..sFT1E%EtFs Nr)next)r;r<non_deprecated_names ` rr6r6s5F F  Q< rc&g}ttj}ttj}t |ds|S||j vr|j t||||j vr|j t|||S)zGet a list of TF 1.* names for this symbol. Args: symbol: symbol to get API names for. Returns: List of all API names for this symbol. r/)r7TENSORFLOW_API_NAMErKERAS_API_NAMEr2r/extendr5)r+names_v1tensorflow_api_attr_v1keras_api_attr_v1s r get_v1_namesrMs('(;<BB">288  $ Ov. OOGF$:;<&//) OOGF$567 /rc&g}ttj}ttj}t |ds|S||j vr|j t||||j vr|j t|||S)zGet a list of TF 2.0 names for this symbol. Args: symbol: symbol to get API names for. Returns: List of all API names for this symbol. r/)r3rGrrHr2r/rIr5)r+names_v2tensorflow_api_attrkeras_api_attrs r get_v2_namesrRs~(!"56<<^,22.  $ OFOO+ OOGF$789v& OOGFN34 /rmodulecg}ttj}t||r|j t |||S)zGet a list of TF 1.* constants in this module. Args: module: TensorFlow module. Returns: List of all API constants under the given module. )r7rGrr2rIr5)rS constants_v1tensorflow_constants_attr_v1s rget_v1_constantsrWsB,!-.A!B!L!L V12(DEF rcg}ttj}t||r|j t |||S)zGet a list of TF 2.0 constants in this module. Args: module: TensorFlow module. Returns: List of all API constants under the given module. )r3rGrr2rIr5)rS constants_v2tensorflow_constants_attrs rget_v2_constantsr[sB,'(;<FF V./(ABC rTc eZdZUdZeeed<eeed<eed<edddded ed eeed e fd Z ddZ de d e fdZ dededeed dfdZdeded dfdZy) api_exportz6Provides ways to export symbols to the TensorFlow API._names _names_v1 _api_nameNT)r,v1allow_multiple_exportsargsr,rbrccV||_||n||_||_|jy)anExport under the names *args (first one is considered canonical). Args: *args: API names in dot delimited format. api_name: API you want to generate Currently, only `tensorflow`. v1: Names for the TensorFlow V1 API. If not set, we will use V2 API names both for TensorFlow V1 and V2 APIs. allow_multiple_exports: Deprecated. N)r_r`ra_validate_symbol_names)selfr,rbrcrds r__init__zapi_export.__init__s+ DK>RtDNDN!rr&ctjtjz}jtk(r.t D]$t fd|Dstdzytfd|Ds tdy)aValidate you are exporting symbols under an allowed package. We need to ensure things exported by tf_export, etc. export symbols under disjoint top-level package names. For TensorFlow, we check that it does not export anything under subpackage names used by components (keras, etc.). For each component, we check that it exports everything under its own subpackage. Raises: InvalidSymbolNameError: If you try to export symbol under disallowed name. c3@K|]}|jywr() startswith)rBn subpackages rrCz4api_export._validate_symbol_names..&sBAq||J'Bsz6@tf_export is not allowed to export symbols under %s.*c3TK|]}|jj!ywr()rkra)rBrlrgs rrCz4api_export._validate_symbol_names..,sH!dnn-Hs%(z8Can only export symbols under package name of component.N) setr_r`rarGSUBPACKAGE_NAMESPACESanyr"all)rgall_symbol_namesrms` @rrfz!api_export._validate_symbol_namess4;;'#dnn*== ~~,,- * B1AB B&F   H7GH H$ F  Irfuncct|jj}t|jj}t j |\}}|j |||j|j |||j|jD] }|t|< |jD]}|td|z<|S)zCalls this decorator. Args: func: decorated symbol (function or class). Returns: The input function with _tf_api_names attribute set. r1) r3rarr7r r4set_attrr_r`r$)rgrtr8api_names_attr_v1r9undecorated_funcr%name_v1s r__call__zapi_export.__call__1st~~.44N$T^^4::&--d3AMM"NDKK@MM"$5t~~F +&*d#+>>?:>nw67? Krr8rct|||yr()setattr)rgrtr8rs rrvzapi_export.set_attrHs D.%(r module_namer%ctj|}t|jj}t |jj}t ||s t||gt||j|j|ft ||s t||gt||j|j|fy)aStore export information for constants/string literals. Export information is stored in the module where constants/string literals are defined. e.g. ```python foo = 1 bar = 2 tf_export("consts.foo").export_constant(__name__, 'foo') tf_export("consts.bar").export_constant(__name__, 'bar') ``` Args: module_name: (string) Name of the module to store constant at. name: (string) Current constant name. N) sysmodulesr3rarr7r2r|r5appendr_r`)rgr}r%rSapi_constants_attrapi_constants_attr_v1s rexport_constantzapi_export.export_constantMs$[[ %F"4>>2<<(8BB 6- . f("- F&'.. T/BC 60 1 f+R0 F)*114>>42HIr)r&N)rrrr#rrrrGrboolrhrfr\rzrrvrrrrr^r^s> 3- c] . *$(%) " "" 8C= ! " # ", :1.))'*)3;C=) ) JJCJDJrr^fcltjfd}tj|S)z3A wrapper that throws away all non-kwarg arguments.cx|r0tdjjjdi|S)Nz`{f} only takes keyword args (possible keys: {kwargs}). Please pass these args as kwargs instead.)rkwargsr) TypeErrorformatrrd)rdrr f_argspecs rwrapperzkwarg_only..wrapperqsB  66>7=7   ;v;r)decorator_argspec)r getfullargspecr make_decorator)rrrs` @r kwarg_onlyrms/''*)  $ $Q9 MMrc 8eZdZddddedeeededefdZy) ExportTypeNT)rbrcv2rbrcr&cyr(r)rgrbrcrs rrzzExportType.__call__s r) rrrrrrrr^rzrrrrr~s> %)%)   8C= !#   rr)r, tf_export keras_export)*r#collections.abcr functoolsrtypingrrrrrtensorflow.python.utilr r rHrGrprr3r7 Exceptionr"dictr$rrr*rr?r6rMrRrWr[r\objectr^rrpartialrrrrrrs2% ??/-"*_6IJK 24JK  2K8  MYM+/&c3h0+s+x}+ (#(2 22!2c] 2j}4