a ==icã@sVddlmZddlZddlZe e¡ZGdd„deƒZeƒZ dd„Z ej dd„ƒZ dS) é)ÚscopeNc@s8eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd S) ÚParameterSharingContextzc This class manages scope driven way of parameter sharing across different NameScopes. cCsi|_g|_dS©N)Ú_scope_overridesÚ _contexts)Úself©rúy/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/caffe2/python/modeling/parameter_sharing.pyÚ__init__sz ParameterSharingContext.__init__cCs‚|}d}| tj¡}d}t|ƒD].\}}||tj}||jvr |j|}|}q ||kr\|S| |¡tj ||dd…¡SdS)aH Recursively resolves all scope overrides, i.e multiple steps of override can be used. For example, if one provides following scope overrides: {'scope_b': 'scope_a'} and within 'scope_b' - {'shared_child': ''}, then name 'w' will get resolved to the following blobs depending on the namescope: a. 'scope_a' -> 'scope_a/w' b. 'scope_b' -> 'scope_a/w' c. 'scope_c' -> 'scope_c/w' d. 'scope_b/shared_child' -> 'scope_a/w' d. 'scope_b/unshared_child' -> 'scope_a/unshared_child/w' rÚéN)ÚsplitrÚ_NAMESCOPE_SEPARATORÚ enumeraterÚ_resolve_scope_overridesÚjoin)rÚcandidate_scopeÚ best_scopeZbest_scope_idxZ sub_scopesZ cur_scopeÚidxZ sub_scoperrr rs    ÿÿz0ParameterSharingContext._resolve_scope_overridescCs4t ¡}| |¡}||kr,t d ||¡¡||S)Nz$Overwriting scope {0} with scope {1})rÚCurrentNameScoperÚloggerÚinfoÚformat)rÚnamerrrrr Úget_parameter_name8s ÿz*ParameterSharingContext.get_parameter_namecCs|j |¡|j |¡dSr)rÚappendrÚupdate)rÚ shared_scopesrrr Úadd_scope_overridesAs z+ParameterSharingContext.add_scope_overridescCs>t|jƒdksJ‚|j ¡i|_|jD]}|j |¡q(dS)Nr)ÚlenrÚpoprr)rÚxrrr r Es   zParameterSharingContext.popN) Ú__name__Ú __module__Ú __qualname__Ú__doc__r rrrr rrrr rs   rcCs$|r|dtjkr|tjS|SdS)Néÿÿÿÿ)rr)Z namescoperrr Ú_normalize_namescopePs r'ccsšt|tƒsJ‚i}t ¡}| ¡D]J\}}| |¡rDJd ||¡ƒ‚||}||}t|ƒ}t|ƒ}|||<q"zt  |¡dVWt  ¡n t  ¡0dS)až Helper function for sharing scopes. All the parameters within the shared_scopes, will be remapped with the respect of CurrentNamescope() I.e. if one calls ParameterSharing with {'scope_b': 'scope_'a'}, from the scope 'some_global_scope', it'll effectively mean, that all parameters from 'some_global_scope/scope_b' will shared with the parameters from 'some_global_scope/scope_a' z:Illegal override for parameter sharing. {} is prefix of {}N) Ú isinstanceÚdictrrÚitemsÚ startswithrr'Úparameter_sharing_contextrr )rZshared_scope_overridesZ current_scopeÚkÚvrrr ÚParameterSharingWs$  ÿÿ  r/) Z caffe2.pythonrÚ contextlibÚloggingÚ getLoggerr"rÚobjectrr,r'Úcontextmanagerr/rrrr Ús  ?