a ==ic'ã@s„ddlZddlZddlmZddlmZdZe ¡add„Z dd„Z ej dd d „ƒZ ej dd d „ƒZ ej dd„ƒZej dd„ƒZdS)éN)Ú basestring)Ú caffe2_pb2ú/cCsttdƒsdt_tjS)NÚ namescopeÚ)ÚhasattrÚ_threadlocal_scoper©r r úd/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/caffe2/python/scope.pyÚCurrentNameScopes r cCsttdƒsdt_tjS)NÚ devicescope)rrr r r r r ÚCurrentDeviceScopes r Fc csŽt|tƒs|dusJdƒ‚tƒ}|r,|tnd}|r<|t_n tj|t_z$dVWtj |¡sfJdƒ‚|t_ntj |¡s‚Jdƒ‚|t_0dS)Nz,NameScope takes in a string as its argument.rzAThe namescope variable is changed from outside NameScope() calls.)Ú isinstancerr Ú_NAMESCOPE_SEPARATORrrÚendswith)ÚprefixÚresetÚ old_scoper r r Ú NameScope#s$ÿ ÿþÿrccsÖt ¡}|r,t|tjƒs Jdƒ‚| |¡n |s8Jdƒ‚|rB||_tƒ}|rh| d¡rh| d¡sh|j|_|r„t|dƒr„|j  |j¡|j  ¡|t _ z"dVWt j |ks°Jdƒ‚|t _ nt j |ksÊJdƒ‚|t _ 0dS)Nz?DeviceScope takes in a caffe2_pb2.DeviceOption as its argument.z7At least one argument should be non-null in DeviceScopeÚ node_nameÚ extra_infoz=The device scope is changed from outside DeviceScope() calls.) rZ DeviceOptionrZCopyFromrr ZHasFieldrrÚextendÚsortrr )ÚscoperZ new_scoperr r r Ú DeviceScope7s8ÿ  ÿ  ÿþ ÿrccs8tƒ}zdt_dVW|t_dS|t_YdS0dS)z© Allow users to 'disable' the name scope behaviour. This sets the CurrentNameScope() to None, so that the field is not set in CreateOperator(...), etc. rN)r rr©rr r r ÚEmptyNameScopeYsÿrccs8tƒ}zdt_dVW|t_dS|t_YdS0dS)a" Allow users to 'disable' the device scope behaviour (so it can be controlled at a NetDef::DeviceOption level, not overridden at OperatorDef::DeviceOption level). This sets the CurrentDeviceScope() to None, so that the field is not set in CreateOperator(...), etc. N)r rr rr r r ÚEmptyDeviceScopejs ÿr)F)N)Ú contextlibÚ threadingZ past.builtinsrZ caffe2.protorrÚlocalrr r Úcontextmanagerrrrrr r r r Ús    !