zj- ddlZddlmZddlmZgZGddZGddZeaea d e d e fd Z d Z dd Z eddZdS)N)deepcopy)signature_safe_contextmanagerc2eZdZdZddZdZdZdZdZdS) UniqueNameGeneratorz Generate unique name with prefix. Args: prefix(str): The generated name prefix. All generated name will be started with this prefix. NcXtjt|_|d}||_dS)N) collections defaultdictintidsprefix)selfrs g/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/paddle/base/unique_name.py__init__zUniqueNameGenerator.__init__ s)*3// >F c,||SN)generate)rkeys r__call__zUniqueNameGenerator.__call__&s}}S!!!rc|j|}|j|xxdz cc<|jd|t|gzS)z Generate unique names with prefix Args: key(str): The key of return string. Returns(str): A unique string with the prefix r_)r rjoinstr)rrtmps rrzUniqueNameGenerator.generate)sJhsm   {SXXsCHHo6666rcddlm}m}|r|S||SNr)_dygraph_tracerin_dygraph_mode) frameworkrr _generate_unique_namer)rrrr s rgenerate_with_ignorable_keyz/UniqueNameGenerator.generate_with_ignorable_key6sV???????? ?   ="?$$::<< <}}S!!!rc`t|j}t|j|_|Sr)rrrr )rrets rclonezUniqueNameGenerator.clone>s'!$+..48$$ rr) __name__ __module__ __qualname____doc__rrrr#r&rrrrsn """ 7 7 7"""rrceZdZdZdZdZdS)DygraphParameterNameCheckerz6 Check whether the name of parameter is used. c,t|_dSr)set _name_set)rs rrz$DygraphParameterNameChecker.__init__IsrcP||jvrdS|j|dS)z Check whether the name is used. If not used, insert into the _name_set. Args: name(str): The name of parameter to check. Returns(bool): If the name is in name_set, return True; Otherwise, return False. TF)r0add)rnames rrz$DygraphParameterNameChecker.__call__Ls1 4> ! !4 N  t $ $ $5rN)r'r(r)r*rrr+rrr-r-Ds<rr-rreturnc t|S)aZ Generate unique name with prefix key. Currently, Paddle distinguishes the names of the same key by numbering it from zero. For example, when key=fc, it continuously generates fc_0, fc_1, fc_2, etc. Args: key(str): The prefix of generated name. Returns: str: A unique string with the prefix key. Examples: .. code-block:: python >>> import paddle >>> name1 = paddle.utils.unique_name.generate('fc') >>> name2 = paddle.utils.unique_name.generate('fc') >>> print(name1, name2) fc_0 fc_1 ) generator)rs rrrbs, S>>rc|ddlm}m}|r|St |Sr)r!rr r"r6)rrr s rr#r#sP;;;;;;;;9  66888 S>>rcrt}t}|tan|a|tan|a||fS)a/ Switch the namespace of in current context to a new namespace. Though :code:`switch()` and :code:`guard()` can both change namespace, :code:`guard()` is recommended since it can manage the context better together with :code:`with` statement. Args: new_generator(UniqueNameGenerator, optional): A new UniqueNameGenerator, not required normally. Default is None, which means switch to a new anonymous namespace. new_para_name_checker(DygraphParameterNameChecker, optional): A new DygraphParameterNameChecker, not required normally. Default is None, which means switch to a new parameter name checker. Returns: UniqueNameGenerator: The previous UniqueNameGenerator. DygraphParameterNameChecker: The previous DygraphParameterNameChecker Examples: .. code-block:: python >>> import paddle >>> name1 = paddle.utils.unique_name.generate('fc') >>> name2 = paddle.utils.unique_name.generate('fc') >>> print(name1, name2) fc_0 fc_1 >>> pre_generator, pre_dygraph_name_checker = paddle.utils.unique_name.switch() # switch to a new anonymous namespace. >>> name2 = paddle.utils.unique_name.generate('fc') >>> print(name2) fc_0 >>> paddle.utils.unique_name.switch(pre_generator, pre_dygraph_name_checker) # switch back to pre_generator. >>> name3 = paddle.utils.unique_name.generate('fc') >>> print(name3) fc_2 )r6dygraph_parameter_name_checkerrr-) new_generatornew_para_name_checker old_generatorold_para_name_checkers rswitchr>sOPM:')) ! $)D)F)F&&)>& / //rc#8Kt|trt|}n6t|tr!t|}t |\}} dVt ||dS#t ||wxYw)a# Change the namespace of unique name with :code:`with` statement. After calling it, a new namespace in the context of :code:`with` will be created, and it will number names from zero again when calling :code:`generate()` with same key. Args: new_generator(str|bytes, optional): New name of global namespace. Note that str in Python2 was split into str and bytes in Python3, so here are two types. Default is None. If not None, new_generator will be added into the prefix of unique name generated by :code:`generate()`. Returns: None. Examples: .. code-block:: python >>> import paddle >>> with paddle.utils.unique_name.guard(): ... name_1 = paddle.utils.unique_name.generate('fc') >>> with paddle.utils.unique_name.guard(): ... name_2 = paddle.utils.unique_name.generate('fc') >>> print(name_1, name_2) fc_0 fc_0 >>> with paddle.utils.unique_name.guard('A'): ... name_1 = paddle.utils.unique_name.generate('fc') >>> with paddle.utils.unique_name.guard('B'): ... name_2 = paddle.utils.unique_name.generate('fc') >>> print(name_1, name_2) Afc_0 Bfc_0 N) isinstancerrbytesdecoder>)r:r<r=s rguardrCsF-%%D+M:: M5 ) )D+M,@,@,B,BCC +1-+@+@(M(5 }344444}34444s 1BB)NNr)r copyrwrapped_decoratorr__all__rr-r9r6rrr#r>rCr+rrrGs<<<<<< ********Z2"=!!>   ! ! ##V40404040n+5+5+5+5+5+5r