o !j\@slddlZddlmZgZGdddZGdddZeaeadd Zd d Z dd d Z edddZ dS)N)signature_safe_contextmanagerc@s"eZdZdZdddZddZdS)UniqueNameGeneratorz Generate unique name with prefix. Args: prefix(str): The generated name prefix. All generated name will be started with this prefix. NcCs"tt|_|dur d}||_dS)N) collections defaultdictintidsprefix)selfr r X/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/base/unique_name.py__init__s  zUniqueNameGenerator.__init__cCs4|j|}|j|d7<|jd|t|gS)z Generate unique names with prefix Args: key(str): The key of return string. Returns(str): A unique string with the prefix r_)r r joinstr)r keytmpr r r __call__%s zUniqueNameGenerator.__call__N__name__ __module__ __qualname____doc__rrr r r r rs  rc@s eZdZdZddZddZdS)DygraphParameterNameCheckerz6 Check whether the name of parameter is used. cCs t|_dSr)set _name_set)r r r r r8s z$DygraphParameterNameChecker.__init__cCs||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)radd)r namer r r r;s  z$DygraphParameterNameChecker.__call__Nrr r r r r3s rcCst|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)rr r r generateQsr!cCs(ddlm}m}|r|St|S)Nr)_dygraph_tracerin_dygraph_mode)Z frameworkr"r#Z_generate_unique_namer )rr"r#r r r generate_with_ignorable_key|s r$cCs>t}t}|dur tan|a|durta||fS|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 N)r dygraph_parameter_name_checkerrr) new_generatorZnew_para_name_checker old_generatorold_para_name_checkerr r r switchs(r)c cs\t|tr t|}n t|trt|}t|\}}z dVWt||dSt||w)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 spilted 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(r r r guards #    r-)NNr) rwrapped_decoratorr__all__rrr%r r!r$r)r-r r r r s + 7